Ejemplos de set_training en Python

Ejemplo n.º 1

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)

Ejemplo n.º 2

Archivo: __init__.py Proyecto: zsk423200/pytorch

def set_training(*args, **kwargs):
    from torch.onnx import utils
    return utils.set_training(*args, **kwargs)

Ejemplo n.º 3

Archivo: __init__.py Proyecto: gtgalone/pytorch

def set_training(*args, **kwargs):
    from torch.onnx import utils
    return utils.set_training(*args, **kwargs)