Add cfg.offline_prioritized_sampler

2024-03-04 23:08:52 +00:00
parent 570f8d01df
commit a027f4edfb
2 changed files with 20 additions and 8 deletions
--- a/lerobot/configs/policy/diffusion.yaml
+++ b/lerobot/configs/policy/diffusion.yaml
@@ -29,6 +29,8 @@ log_freq: 250
 offline_steps: 1344000
 online_steps: 0

+offline_prioritized_sampler: true
+
 policy:
  name: diffusion