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越来越小,准备度越来越高,代表训练正常进行。
可随意转载,欢迎署名!
微信扫一扫
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越来越小,准备度越来越高,代表训练正常进行。
可随意转载,欢迎署名!
相关推荐