insightface mxnet 神经网络人脸识别训练

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越来越小,准备度越来越高,代表训练正常进行。

posted @ 2023-02-16 14:35  忙碌在路上  阅读(41)  评论(0编辑  收藏  举报