tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Checkpoint on final step of training even when it doesn't coincide with save_freq. (#284)

Browse Source
This commit is contained in:
Alexander Soare
2024-06-20 08:27:01 +01:00
committed by GitHub
parent 2abef3bef9
commit 9aa4cdb976
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/configs/default.yaml
View File

@@ -39,9 +39,10 @@ training:
# `online_env_seed` is used for environments for online training data rollouts.
online_env_seed: ???
eval_freq: ???
save_freq: ???
log_freq: 250
save_checkpoint: true
# Checkpoint is saved every `save_freq` training iterations and after the last training step.
save_freq: ???
num_workers: 4
batch_size: ???
image_transforms: