early training loss as expected

2024-03-11 13:34:04 +00:00
parent fab2b3240b
commit 2a01487494
5 changed files with 232 additions and 8 deletions
--- a/lerobot/configs/policy/diffusion.yaml
+++ b/lerobot/configs/policy/diffusion.yaml
@@ -84,9 +84,9 @@ obs_encoder:
  imagenet_norm: True

 rgb_model:
-  _target_: lerobot.common.policies.diffusion.pytorch_utils.get_resnet
-  name: resnet18
-  weights: null
+  model_name: resnet18
+  pretrained: false
+  num_keypoints: 32

 ema:
  _target_: lerobot.common.policies.diffusion.model.ema_model.EMAModel