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 神经网络人脸识别训练
阅读 108
2023-03-26
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越来越小,准备度越来越高,代表训练正常进行。
可随意转载,欢迎署名!
相关推荐
精彩评论(0)