wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add message

This commit is contained in:
Remi Cadene 2024-05-20 10:08:58 +00:00
parent 89393b505b
commit a0f23e9176
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/scripts/save_policy_to_safetensor.py
View File

@ -92,6 +92,9 @@ def save_policy_to_safetensors(output_dir, env_name, policy_name, extra_override
env_policy_dir = Path(output_dir) / f"{env_name}_{policy_name}" env_policy_dir = Path(output_dir) / f"{env_name}_{policy_name}"
if env_policy_dir.exists(): if env_policy_dir.exists():
print(f"Overwrite existing safetensors in '{env_policy_dir}':")
print(f" - Validate with: `git add {env_policy_dir}`")
print(f" - Revert with: `git checkout -- {env_policy_dir}`")
shutil.rmtree(env_policy_dir) shutil.rmtree(env_policy_dir)
env_policy_dir.mkdir(parents=True, exist_ok=True) env_policy_dir.mkdir(parents=True, exist_ok=True)