We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
map_location=device
1 parent 5e94738 commit 25c63ccCopy full SHA for 25c63cc
examples/2_evaluate_pretrained_policy.py
@@ -44,7 +44,7 @@
44
# OR a path to a local outputs/train folder.
45
# pretrained_policy_path = Path("outputs/train/example_pusht_diffusion")
46
47
-policy = DiffusionPolicy.from_pretrained(pretrained_policy_path, map_location=device)
+policy = DiffusionPolicy.from_pretrained(pretrained_policy_path)
48
49
# Initialize evaluation environment to render two observation types:
50
# an image of the scene and state/position of the agent. The environment
0 commit comments