verify t5 inference and no backward crush
Showing
- SwissArmyTransformer/arguments.py 2 additions, 0 deletionsSwissArmyTransformer/arguments.py
- SwissArmyTransformer/model/t5_model.py 6 additions, 3 deletionsSwissArmyTransformer/model/t5_model.py
- SwissArmyTransformer/mpu/transformer.py 3 additions, 6 deletionsSwissArmyTransformer/mpu/transformer.py
- SwissArmyTransformer/tokenization/hf_tokenizer.py 1 addition, 1 deletionSwissArmyTransformer/tokenization/hf_tokenizer.py
- examples/t5/config/config_t5_large.json 1 addition, 4 deletionsexamples/t5/config/config_t5_large.json
- examples/t5/config/model_t5_large.sh 1 addition, 2 deletionsexamples/t5/config/model_t5_large.sh
- examples/t5/finetune_t5.py 83 additions, 0 deletionsexamples/t5/finetune_t5.py
- examples/t5/inference_t5.py 25 additions, 164 deletionsexamples/t5/inference_t5.py
- examples/t5/scripts/config_t5_tmp.json 23 additions, 0 deletionsexamples/t5/scripts/config_t5_tmp.json
- examples/t5/scripts/finetune_t5.sh 54 additions, 0 deletionsexamples/t5/scripts/finetune_t5.sh
- examples/t5/scripts/generate_t5.sh 4 additions, 5 deletionsexamples/t5/scripts/generate_t5.sh
- examples/t5/test_t5.py 5 additions, 8 deletionsexamples/t5/test_t5.py
- setup.py 1 addition, 1 deletionsetup.py
Loading
Please register or sign in to comment