admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-23 20:50:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix path optimizer state

Browse Source
This commit is contained in:
Pepijn
2025-10-14 11:26:57 +02:00
parent 6486982ab4
commit a86cea5708
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/training/test_multi_gpu.py
+1 -1
View File
@@ -218,5 +218,5 @@ class TestMultiGPUTraining:
assert training_state_dir.exists(), f"No training state in checkpoint {checkpoint_dir}" assert training_state_dir.exists(), f"No training state in checkpoint {checkpoint_dir}"
# Verify optimizer state exists # Verify optimizer state exists
optimizer_state = training_state_dir / "optimizer_state.pt" optimizer_state = training_state_dir / "optimizer_state.safetensors"
assert optimizer_state.exists(), f"No optimizer state in checkpoint {checkpoint_dir}" assert optimizer_state.exists(), f"No optimizer state in checkpoint {checkpoint_dir}"