Save costs training on AMLCompute