506821c7df
- added `torch.compile` to the actor and learner servers. - organized imports in `train_sac.py` - optimized the parameters push by not sending the frozen pre-trained encoder. Co-authored-by: Adil Zouitine <adilzouitinegm@gmail.com> |
||
|---|---|---|
| .. | ||
| act.yaml | ||
| act_aloha_real.yaml | ||
| act_koch_real.yaml | ||
| act_moss_real.yaml | ||
| act_so100_real.yaml | ||
| diffusion.yaml | ||
| diffusion_pusht_keypoints.yaml | ||
| hilserl_classifier.yaml | ||
| sac_maniskill.yaml | ||
| sac_pusht_keypoints.yaml | ||
| tdmpc.yaml | ||
| tdmpc_pusht_keypoints.yaml | ||
| vqbet.yaml | ||