CyberDJ commited on
Commit
6e7cc4b
·
verified ·
1 Parent(s): 734ef1d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -4,4 +4,6 @@ license: mit
4
  The following pi0-fast weights were obtained by training on 4 A100 GPUs for 10k iterations using five tasks (2500 episodes) from the primitive-ft-dataset, shared for community reference.
5
 
6
  The five primitive tasks used to train are: [select_fruit, select_toy, select_painting, select_poker, select_mahjong].
7
- These tasks involve similar skills and simple actions, making them suitable for research on downstream adaptation and generalization abilities.
 
 
 
4
  The following pi0-fast weights were obtained by training on 4 A100 GPUs for 10k iterations using five tasks (2500 episodes) from the primitive-ft-dataset, shared for community reference.
5
 
6
  The five primitive tasks used to train are: [select_fruit, select_toy, select_painting, select_poker, select_mahjong].
7
+ These tasks involve similar skills and simple actions, making them suitable for research on downstream adaptation and generalization abilities.
8
+
9
+ The training codes are available at: https://github.com/Shiduo-zh/openpi. If any issues or bugs are encountered during training, feel free to contact our team.