diff --git a/cnn_benchmark/config.py b/Classification/resnet50v1.5/config.py similarity index 100% rename from cnn_benchmark/config.py rename to Classification/resnet50v1.5/config.py diff --git a/cnn_benchmark/job_function_util.py b/Classification/resnet50v1.5/job_function_util.py similarity index 100% rename from cnn_benchmark/job_function_util.py rename to Classification/resnet50v1.5/job_function_util.py diff --git a/cnn_benchmark/of_cnn_train_val.py b/Classification/resnet50v1.5/of_cnn_train_val.py similarity index 100% rename from cnn_benchmark/of_cnn_train_val.py rename to Classification/resnet50v1.5/of_cnn_train_val.py diff --git a/cnn_benchmark/of_cnn_val.py b/Classification/resnet50v1.5/of_cnn_val.py similarity index 100% rename from cnn_benchmark/of_cnn_val.py rename to Classification/resnet50v1.5/of_cnn_val.py diff --git a/cnn_benchmark/ofrecord_util.py b/Classification/resnet50v1.5/ofrecord_util.py similarity index 100% rename from cnn_benchmark/ofrecord_util.py rename to Classification/resnet50v1.5/ofrecord_util.py diff --git a/cnn_benchmark/optimizer_util.py b/Classification/resnet50v1.5/optimizer_util.py similarity index 100% rename from cnn_benchmark/optimizer_util.py rename to Classification/resnet50v1.5/optimizer_util.py diff --git a/cnn_benchmark/resnet_model.py b/Classification/resnet50v1.5/resnet_model.py similarity index 100% rename from cnn_benchmark/resnet_model.py rename to Classification/resnet50v1.5/resnet_model.py diff --git a/Classification/resnet50v1.5/train.sh b/Classification/resnet50v1.5/train.sh new file mode 100755 index 0000000000000000000000000000000000000000..38c627203e27c319afc089b6764741a9a092bbf6 --- /dev/null +++ b/Classification/resnet50v1.5/train.sh @@ -0,0 +1,20 @@ +export ENABLE_USER_OP=True +rm -rf core.* +rm -rf ./output/snapshots/* + +DATA_ROOT=/dataset/ImageNet/ofrecord + +python3 of_cnn_train_val.py \ + --train_data_dir=$DATA_ROOT/train \ + --train_data_part_num=256 \ + --val_data_dir=$DATA_ROOT/validation \ + --val_data_part_num=256 \ + --num_nodes=1 \ + --gpu_num_per_node=4 \ + --model_update="momentum" \ + --learning_rate=0.256 \ + --loss_print_every_n_iter=20 \ + --batch_size_per_device=64 \ + --val_batch_size_per_device=125 \ + --num_epoch=90 \ + --model="resnet50" \ No newline at end of file diff --git a/cnn_benchmark/util.py b/Classification/resnet50v1.5/util.py similarity index 100% rename from cnn_benchmark/util.py rename to Classification/resnet50v1.5/util.py diff --git a/run_cnn_train.sh b/run_cnn_train.sh deleted file mode 100755 index e5c8c6f12959e842fd4acfb4742a8c3e7fa281c1..0000000000000000000000000000000000000000 --- a/run_cnn_train.sh +++ /dev/null @@ -1,29 +0,0 @@ -export ENABLE_USER_OP=True -rm -rf core.* -rm -rf ./output/snapshots/* - -# DATA_ROOT=/dataset/ImageNet/ofrecord - -python3 cnn_benchmark/of_cnn_train_val.py \ - --train_data_part_num=256 \ - --val_data_part_num=256 \ - --num_nodes=1 \ - --gpu_num_per_node=1 \ - --model_update="momentum" \ - --mom=0.875 \ - --learning_rate=0.256 \ - --loss_print_every_n_iter=1 \ - --batch_size_per_device=8 \ - --val_batch_size_per_device=8 \ - --num_epoch=90 \ - --use_fp16=false \ - --use_boxing_v2=true \ - --model="resnet50" - - # --node_ips='11.11.1.12,11.11.1.14' \ - # --model_load_dir=/DATA/disk1/mx_imagenet/of_mx_mixed_init_model \ - # --train_data_dir=$DATA_ROOT/train \ - # --train_data_part_num=256 \ - # --val_data_dir=$DATA_ROOT/validation \ - # --val_data_part_num=256 \ - #--num_examples=1024 \