ResNet50.sh 447 字节
Newer Older
R
ruri 已提交
1 2
##Training details
#GPU: NVIDIA® Tesla® V100 4cards 120epochs 67h
L
Leo Chen 已提交
3
export CUDA_VISIBLE_DEVICES=0
R
ruri 已提交
4 5 6 7 8 9 10
export FLAGS_fast_eager_deletion_mode=1
export FLAGS_eager_delete_tensor_gb=0.0
export FLAGS_fraction_of_gpu_memory_to_use=0.98

#ResNet50:
python train.py \
       --model=ResNet50 \
L
Leo Chen 已提交
11
       --batch_size=64 \
R
ruri 已提交
12 13 14 15 16
       --model_save_dir=output/ \
       --lr_strategy=piecewise_decay \
       --num_epochs=120 \
       --lr=0.1 \
       --l2_decay=1e-4