torch.onnx — PyTorch 1.10.1 documentation
https://pytorch.org/docs/stable/onnx.htmltorch.onnx. select_model_mode_for_export (model, mode) [source] ¶ A context manager to temporarily set the training mode of model to mode, resetting it when we exit the with-block. A no-op if mode is None. Parameters. model – Same type and meaning as model arg to export(). mode – Same type and meaning as training arg to export().