train\_mode =========== .. currentmodule:: mxnet.autograd .. autofunction:: train_mode