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
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