def set_training(model, mode): r""" 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. """ from torch.onnx import utils return utils.set_training(model, mode)
def set_training(*args, **kwargs): from torch.onnx import utils return utils.set_training(*args, **kwargs)
def set_training(*args, **kwargs): from torch.onnx import utils return utils.set_training(*args, **kwargs)