diff --git a/src/super_gradients/recipes/imagenet_regnetY.yaml b/src/super_gradients/recipes/imagenet_regnetY.yaml index 3999bdd5f5..e101b5d813 100644 --- a/src/super_gradients/recipes/imagenet_regnetY.yaml +++ b/src/super_gradients/recipes/imagenet_regnetY.yaml @@ -47,7 +47,7 @@ training_hyperparams: ckpt_root_dir: multi_gpu: Off - +num_gpus: 1 architecture: regnetY800 experiment_name: ${architecture}