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越来越小,准备度越来越高,代表训练正常进行。
insightface mxnet 神经网络人脸识别训练
https://blog.coolsre.com/archives/insightface-mxnet-shen-jing-wang-luo-ren-lian-shi-bie-xun-lian
评论