#Training details #GPU: NVIDIA® Tesla® V100 4cards 200epochs 120h export CUDA_VISIBLE_DEVICES=0,1,2,3 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 \
--data_dir=./data/ILSVRC2012/ \ --total_images=1281167 \ --class_dim=1000 \ --validate=1 \ --model=ResNet50_vd \ --batch_size=256 \ --lr_strategy=cosine_decay \ --lr=0.1 \ --num_epochs=200 \ --model_save_dir=output/ \ --l2_decay=7e-5 \ --use_mixup=True \ --use_label_smoothing=True \ --label_smoothing_epsilon=0.1
