mbreuss commited on
Commit
026f834
1 Parent(s): c4717ec

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -16,6 +16,9 @@ pipeline_tag: robotics
16
 
17
  ## Model Description
18
 
 
 
 
19
  This model implements a Mixture of Diffusion Experts architecture for robotic manipulation, combining transformer-based backbone with noise-only expert routing. For faster inference, we can precache the chosen expert for each timestep to reduce computation time.
20
 
21
  The model has been pretrained on a subset of OXE for 300k steps and finetuned for downstream tasks on the CALVIN/LIBERO dataset.
@@ -57,8 +60,8 @@ action = model.step(obs, goal)
57
 
58
  ### Configuration
59
  - **Optimizer**: AdamW
60
- - **Learning Rate**: {1e-4}
61
- - **Weight Decay**: {0.05}
62
 
63
  ## License
64
  This model is released under the MIT license.
 
16
 
17
  ## Model Description
18
 
19
+ - [Github Link](https://github.com/intuitive-robots/MoDE_Diffusion_Policy)
20
+ - [Project Page](https://mbreuss.github.io/MoDE_Diffusion_Policy/)
21
+
22
  This model implements a Mixture of Diffusion Experts architecture for robotic manipulation, combining transformer-based backbone with noise-only expert routing. For faster inference, we can precache the chosen expert for each timestep to reduce computation time.
23
 
24
  The model has been pretrained on a subset of OXE for 300k steps and finetuned for downstream tasks on the CALVIN/LIBERO dataset.
 
60
 
61
  ### Configuration
62
  - **Optimizer**: AdamW
63
+ - **Learning Rate**: 0.0001
64
+ - **Weight Decay**: 0.05
65
 
66
  ## License
67
  This model is released under the MIT license.