We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 20e7d9b commit 9025172Copy full SHA for 9025172
1 file changed
deepmd/pt/infer/deep_eval.py
@@ -130,7 +130,7 @@ def __init__(
130
] = state_dict[item].clone()
131
state_dict = state_dict_head
132
model = get_model(self.input_param).to(DEVICE)
133
- model = torch.jit.script(model)
+ # model = torch.jit.script(model)
134
self.dp = ModelWrapper(model)
135
self.dp.load_state_dict(state_dict)
136
elif str(self.model_path).endswith(".pth"):
0 commit comments