提交 f246c2e9 编写于 作者: Z zhengya01

add ce

上级 7b7385e3
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册