admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-06 17:41:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

more eval fixes

Browse Source
This commit is contained in:
Jade Choghari
2025-11-16 11:22:05 +01:00
parent cde2e24d79
commit 589788e760
2 changed files with 51 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_3.py
+2 -1
View File
@@ -68,6 +68,7 @@ inputs.update({
"domain_id": domain_id.to("cuda"),
})
breakpoint()
for k in inputs.keys() & inputs_1.keys(): # intersection of keys
a = inputs[k]
b = inputs_1[k].to("cuda")
@@ -100,7 +101,7 @@ action = model.generate_actions(**inputs, steps=10).squeeze(0).float().cpu().num
action_1 = policy.model.generate_actions(**inputs, steps=10).squeeze(0).float().cpu().numpy()
#np all close
print(np.allclose(action, action_1, atol=1e-4, rtol=1e-4))
print(np.allclose(action, action_1, atol=1e-2, rtol=1e-2))
print("max diff:", np.max(np.abs(action - action_1)))
print("mean diff:", np.mean(np.abs(action - action_1)))
breakpoint()