提交 f6268dbd 编写于 作者: S ShawnXuan

boxing v2 and new dataloader

上级 6999f340
rm -rf core.* rm -rf core.*
rm -rf ./output/snapshots/* rm -rf ./output/snapshots/*
#DATA_ROOT=/mnt/13_nfs/xuan/ImageNet/ofrecord #DATA_ROOT=/DATA/disk1/of_imagenet_example
DATA_ROOT=/dataset/ImageNet/ofrecord DATA_ROOT=/DATA/disk1/ImageNet/ofrecord
#DATA_ROOT=/dataset/ImageNet/ofrecord
#DATA_ROOT=/dataset/imagenet-mxnet #DATA_ROOT=/dataset/imagenet-mxnet
#python3 cnn_benchmark/of_cnn_train_val.py \ #python3 cnn_benchmark/of_cnn_train_val.py \
#nvprof -f -o resnet.nvvp \ #nvprof -f -o resnet.nvvp \
...@@ -13,13 +14,19 @@ DATA_ROOT=/dataset/ImageNet/ofrecord ...@@ -13,13 +14,19 @@ DATA_ROOT=/dataset/ImageNet/ofrecord
--val_data_part_num=256 \ --val_data_part_num=256 \
--num_nodes=1 \ --num_nodes=1 \
--node_ips='11.11.1.12,11.11.1.14' \ --node_ips='11.11.1.12,11.11.1.14' \
--gpu_num_per_node=4 \ --gpu_num_per_node=8 \
--optimizer="momentum-cosine-decay" \ --optimizer="momentum-cosine-decay" \
--learning_rate=0.256 \ --learning_rate=0.256 \
--loss_print_every_n_iter=20 \ --loss_print_every_n_iter=20 \
--batch_size_per_device=32 \ --batch_size_per_device=64 \
--val_batch_size_per_device=125 \ --val_batch_size_per_device=125 \
--use_boxing_v2=True \
--use_new_dataloader=True \
--model="resnet50" --model="resnet50"
# --train_data_dir=$DATA_ROOT/train \
# --train_data_part_num=256 \
# --val_data_dir=$DATA_ROOT/validation \
# --val_data_part_num=256 \
#--use_fp16 true \ #--use_fp16 true \
#--weight_l2=3.0517578125e-05 \ #--weight_l2=3.0517578125e-05 \
#--num_examples=1024 \ #--num_examples=1024 \
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册