一般情况下,不需要修改有关环境变量的配置。但是一些特殊情况,需要修改的,就涉及到以下这些内容了:
在linux上最简单的修改方式就是export MXNET_GPU_WORKER_NTHREADS=3
MXNET_BACKWARD_DO_MIRROR=1
将会节省30%~50%的显存, 保存95%的运行速度。更多配置信息:
http://mxnet.incubator.apache.org/how_to/env_var.html
Mxnet可以通过断点恢复训练,命令如下:
python train_end2end.py --network resnet --gpu 0 --resume --begin_epoch 10 --end_epoch 15 --lr_step 15