diff --git a/fluid/PaddleCV/video_classification/.run_ce.sh b/fluid/PaddleCV/video_classification/.run_ce.sh index cfbe248b4d75f0daed55f92aafb75063fa20266c..2461044aad4395912aba8e57526cd3d51c25c397 100755 --- a/fluid/PaddleCV/video_classification/.run_ce.sh +++ b/fluid/PaddleCV/video_classification/.run_ce.sh @@ -9,7 +9,7 @@ export CUDA_VISIBLE_DEVICES=$cudaid export FLAGS_fraction_of_gpu_memory_to_use=0.5 FLAGS_benchmark=true python train.py --batch_size=16 --total_videos=9537 --class_dim=101 --num_epochs=1 --image_shape=3,224,224 --model_save_dir=output/ --with_mem_opt=True --lr_init=0.01 --num_layers=50 --seg_num=7 --enable_ce=True | python _ce.py -#export FLAGS_fraction_of_gpu_memory_to_use=0.92 +export FLAGS_fraction_of_gpu_memory_to_use=0.92 cudaid=${video_classification_4:=0,1,2,3} # use 0,1,2,3 card as default