For training data and checkpoints, please refer to Data & Model Checkpoints. Please fill in OpenAI key in evaluate.py for online gpt access.
| LanguageType | Meaning |
|---|---|
| no | No Language |
| h | Hindsight |
| f | Foresight |
| hf | Hindsight & Foresight |
| rh | GPT-augmented Hindsight |
| rf | GPT-augmented Foresight |
| rhf | GPT-augmented Hindsight & Foresight |
source env.sh
source scripts/train/h1/train_{LanguageType}.sh
source env.sh
source scripts/train/h2/pretrain/pretrain_{LanguageType}.sh
Please modify the load_ckpt path and the shot_number in the script.
source env.sh
source scripts/train/h2/adapt.sh
source env.sh
source scripts/eval/eval_h1.sh
source env.sh
source scripts/eval/eval_h2.sh