Update eval_configs.py

ilyes319 · ilyes319 · commit d1b2a282330e · 2022-02-18T11:33:43.000+01:00
diff --git a/scripts/eval_configs.py b/scripts/eval_configs.py
@@ -54,7 +54,7 @@ def main():
     )
 
     # Load model
-    model: modules.BodyOrderedModel = torch.load(f=args.model, map_location=device)
+    model = torch.load(f=args.model, map_location=device)
 
     # Collect data
     energies_list = []

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,7 @@ def main():`
`54`	`54`	`)`
`55`	`55`
`56`	`56`	`# Load model`
`57`		`- model: modules.BodyOrderedModel = torch.load(f=args.model, map_location=device)`
	`57`	`+ model = torch.load(f=args.model, map_location=device)`
`58`	`58`
`59`	`59`	`# Collect data`
`60`	`60`	`energies_list = []`