Finetuning the v2 version of the base model
- Langage dominant
- Python
- Étoiles
- 441
- Forks
- 44
- Métriques de merge des PR
- Aucune PR mergée en 30 j
Description
Hi,
I am trying to finetune the v2 version of the base model, downloaded from [here](https://console.cloud.google.com/storage/browser/unifiedqa/models_v2/base?pageState=(%22StorageObjectListTable%22:(%22f%22:%22%255B%255D%22))&prefix=&forceOnObjectsSortingFiltering=false).
This is the command I am using,
```
t5_mesh_transformer \
--model_dir="/tmp/model_out" \
--gin_param="utils.run.mesh_devices = ['gpu:0','gpu:1']" \
--gin_param="utils.run.train_dataset_fn = @t5.models.mesh_transformer.tsv_dataset_fn" \
--gin_param="utils.run.mesh_shape = 'model:1,batch:2'" \
--gin_param="tsv_dataset_fn.filename = 'train.tsv'" \
--gin_file="operative_config.gin" \
--gin_param="run.train_steps = 1260900"
```
I am using the the initial checkpoint as (in the operative_config.gin) file ,
`init_checkpoint = `/path/to/downloaded/v2/base/model.ckpt-1250900'`
and I want to train for 10000 epochs, hence `--gin_param="run.train_steps = 1260900"`
Given the task, is this the right setup?
I am seeing poor performance, max 8% accuracy after the training step.
I am attaching the `train.tsv` and the `operative_config.gin` files.
[Archive.zip](https://github.com/allenai/unifiedqa/files/9163322/Archive.zip)
Any help is appreciated. Thanks
Guide de contribution
Aucun guide de contribution indexé pour ce dépôt
Évaluation
Cette issue n'a pas encore été évaluée.