CUDA_VISIBLE_DEVICES='0' MXNET_CUDNN_LIB_CHECKING=0 MXNET_USE_FUSION=0 MXNET_CUDNN_AUTOTUNE_DEFAULT=0 python -u train.py --network r100 --dataset emore --ckpt 2 --per-batch-size 40

备注:可以适当调整per-batch-size大小,当loss越来越小,准备度越来越高,代表训练正常进行。

可随意转载,欢迎署名!