Exemplos de Net.eval em Python

Exemplo n.º 1

Arquivo: 2_test_rule-base-simulator_nlg-template.py Projeto: qbetterk/user-simulator

def load_policy_model(model_dir="model/test_nlg_no_warm_up_with_nlu.pkl"):
    # f.seek(0)
    model = torch.load(model_dir)
    net = Net(state_dim=dialog_config.STATE_DIM, num_actions=dialog_config.SYS_ACTION_CARDINALITY, config=config).to(device)
    net.load_state_dict(model)
    net.eval()
    return net

Exemplo n.º 2

def load_policy_model(model_dir):
    model = torch.load(model_dir)
    net = Net(state_dim=dialog_config.STATE_DIM,
              num_actions=dialog_config.SYS_ACTION_CARDINALITY,
              config=config).to(device)
    net.load_state_dict(model)
    net.eval()
    return net