torch:train

(torch:train module &optional mode)

モジュールとすべてのサブモジュールを学習モード (PyTorch の nn.Module.train) にし、モジュール自身を返します。mode に明示的に nil を渡すと torch:eval と同じです。現在このフラグを読むのは torch:dropout だけです。