diff --git a/cv/classification/resnet50/tensorflow/run_train_resnet50_multigpu_imagenette.sh b/cv/classification/resnet50/tensorflow/run_train_resnet50_multigpu_imagenette.sh index c4ab33fa3f8e840c4d75e295c9069242eed33efb..1a2c33ad7ab84a1078f9eeaeb94522c72591e797 100644 --- a/cv/classification/resnet50/tensorflow/run_train_resnet50_multigpu_imagenette.sh +++ b/cv/classification/resnet50/tensorflow/run_train_resnet50_multigpu_imagenette.sh @@ -42,7 +42,7 @@ done source ./get_num_devices.sh UMD_WAITAFTERLAUNCH=1 python3 -u tf_cnn_benchmarks.py\ - --data_name=imagenet --data_dir=${DATA_DIR}\ + --data_name=imagenette --data_dir=${DATA_DIR}\ --data_format=NCHW --batch_size=${BATCH_SIZE}\ --model=resnet50 --optimizer=${OPTIMIZER} --num_gpus=${IX_NUM_CUDA_VISIBLE_DEVICES}\ --weight_decay=1e-4 --train_dir=${TRAIN_DIR}\