ResNet18_vd.sh 526 字节
Newer Older
1
export CUDA_VISIBLE_DEVICES=0,1,2,3
R
ruri 已提交
2 3 4 5 6
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

python train.py \
7
            --model=ResNet18_vd \
R
ruri 已提交
8 9 10 11 12
            --batch_size=256 \
            --lr_strategy=cosine_decay \
            --lr=0.1 \
            --num_epochs=200 \
            --model_save_dir=output/ \
13
            --l2_decay=7e-5 \
R
ruri 已提交
14 15
            --use_mixup=True \
            --use_label_smoothing=True \
16
            --label_smoothing_epsilon=0.1