allenai/bhaskara

Model

should probably proofread and complete it, then remove this comment. -->

14

5 commits

2 linked in READMEs

updated Mar 6, 2023

See the code

README

output

Model description

This model is a fine-tuned version of EleutherAI/gpt-neo-2.7B on the Lila-IID-train/dev set from the Lila dataset.

Usage

Bhaskara was trained with the following format:

Question: ...

Answer: ...

Program:
```python
...
```

It will perform best if queried in this way.

Intended uses & limitations

If you use this model, please cite our work.

@INPROCEEDINGS{Mishra2022Lila,
  author = {
    Swaroop Mishra 
      and Matthew Finlayson 
      and Pan Lu 
      and Leonard Tang 
      and Sean Welleck 
      and Chitta Baral 
      and Tanmay Rajpurohit 
      and Oyvind Tafjord 
      and Ashish Sabharwal 
      and Peter Clark 
      and Ashwin Kalyan},
  title = {Lila: A Unified Benchmark for Mathematical Reasoning},
  booktitle = {Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing (EMNLP)},
  year = {2022}
}

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • distributed_type: multi-GPU
  • num_devices: 2
  • total_train_batch_size: 8
  • total_eval_batch_size: 8
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 10.0

Training results

Training LossEpochStepValidation LossAccuracy
No log0.061000.79300.8214
No log0.112000.75440.8290
No log0.173000.73580.8328
No log0.234000.71920.8357
0.81560.285000.70120.8397
0.81560.346000.69040.8419
0.81560.47000.68020.8440
0.81560.458000.66700.8465
0.81560.519000.65720.8486
0.72190.5710000.64990.8500
0.72190.6211000.64110.8522
0.72190.6812000.63430.8537
0.72190.7413000.62990.8546
0.72190.7914000.62210.8561
0.6620.8515000.61570.8574
0.6620.9116000.61380.8579
0.6620.9617000.60550.8595
0.6621.0218000.61430.8598
0.6621.0819000.61910.8599
0.57071.1420000.61180.8607
0.57071.1921000.61230.8611
0.57071.2522000.60890.8617
0.57071.3123000.60640.8619
0.57071.3624000.60790.8625
0.49231.4225000.60400.8625
0.49231.4826000.60300.8630
0.49231.5327000.60210.8636
0.49231.5928000.60010.8643
0.49231.6529000.59810.8644
0.49091.730000.59420.8648
0.49091.7631000.59180.8650
0.49091.8232000.59230.8659
0.49091.8733000.58840.8664
0.49091.9334000.58840.8663
0.49641.9935000.59030.8669
0.49642.0436000.64210.8655
0.49642.137000.64010.8651
0.49642.1638000.64110.8649
0.49642.2139000.63870.8645
0.3452.2740000.63620.8654
0.3452.3341000.63620.8654
0.3452.3842000.63620.8654
0.3452.4443000.63570.8655
0.3452.544000.63620.8656
0.34632.5545000.63770.8658
0.34632.6146000.63570.8660
0.34632.6747000.62940.8665
0.34632.7248000.63330.8665
0.34632.7849000.63620.8662
0.35082.8450000.63570.8666
0.35082.8951000.62990.8673
0.35082.9552000.63130.8668
0.35083.0153000.71880.8646
0.35083.0654000.70170.8656
0.2953.1255000.69820.8653
0.2953.1856000.70310.8655
0.2953.2357000.69920.8651
0.2953.2958000.69970.8653
0.2953.3559000.70410.8651
0.23483.4160000.70750.8649
0.23483.4661000.69920.8650
0.23483.5262000.70650.8647
0.23483.5863000.69970.8652
0.23483.6364000.70260.8651
0.24113.6965000.70460.8656
0.24113.7566000.70070.8655
0.24113.867000.70260.8651
0.24113.8668000.70310.8655
0.24113.9269000.70120.8658
0.2513.9770000.70510.8656
0.2514.0371000.76070.8650
0.2514.0972000.76320.8656
0.2514.1473000.75880.8655
0.2514.274000.75780.8651
0.17974.2675000.77100.8645
0.17974.3176000.76270.8648
0.17974.3777000.75830.8650
0.17974.4378000.76460.8649
0.17974.4879000.75980.8646
0.17844.5480000.76560.8650
0.17844.681000.76170.8648
0.17844.6582000.75730.8651
0.17844.7183000.76710.8648
0.17844.7784000.75630.8651
0.18274.8285000.76510.8649
0.18274.8886000.76370.8650
0.18274.9487000.76070.8654
0.18274.9988000.76070.8650
0.18275.0589000.81490.8646
0.1675.1190000.80810.8648
0.1675.1691000.81840.8644
0.1675.2292000.81400.8647
0.1675.2893000.81690.8644
0.1675.3394000.81200.8645
0.13715.3995000.81540.8643
0.13715.4596000.81790.8642
0.13715.5197000.81540.8643
0.13715.5698000.81200.8645
0.13715.6299000.81100.8650
0.14255.68100000.81590.8645
0.14255.73101000.81740.8646
0.14255.79102000.81590.8649
0.14255.85103000.81100.8639
0.14255.9104000.81350.8645
0.15055.96105000.81400.8642
0.15056.02106000.86280.8640
0.15056.07107000.85400.8644
0.15056.13108000.85300.8642
0.15056.19109000.85600.8647
0.10866.24110000.85550.8649
0.10866.3111000.86040.8644
0.10866.36112000.85690.8642
0.10866.41113000.85300.8639
0.10866.47114000.85890.8643
0.10766.53115000.85250.8639
0.10766.58116000.85790.8640
0.10766.64117000.85940.8640
0.10766.7118000.85990.8643
0.10766.75119000.85640.8640
0.11096.81120000.86330.8640
0.11096.87121000.85840.8638
0.11096.92122000.86470.8636
0.11096.98123000.85990.8635
0.11097.04124000.89790.8632
0.10287.09125000.89360.8635
0.10287.15126000.90430.8637
0.10287.21127000.89890.8642
0.10287.26128000.89360.8642
0.10287.32129000.89210.8641
0.07747.38130000.89550.8634
0.07747.43131000.89500.8636
0.07747.49132000.89940.8635
0.07747.55133000.89990.8635
0.07747.6134000.89360.8631
0.08527.66135000.90480.8634
0.08527.72136000.89600.8632
0.08527.78137000.90230.8635
0.08527.83138000.89840.8638
0.08527.89139000.90190.8635
0.08797.95140000.90140.8634
0.08798.0141000.91360.8630
0.08798.06142000.93120.8639
0.08798.12143000.93460.8635
0.08798.17144000.93070.8635
0.06118.23145000.94190.8641
0.06118.29146000.93310.8631
0.06118.34147000.93750.8636
0.06118.4148000.92920.8626
0.06118.46149000.94580.8637
0.0618.51150000.93360.8634
0.0618.57151000.94090.8630
0.0618.63152000.93900.8632
0.0618.68153000.93750.8628
0.0618.74154000.93650.8630
0.06468.8155000.93700.8628
0.06468.85156000.93550.8629
0.06468.91157000.93750.8632
0.06468.97158000.93900.8630
0.06469.02159000.97170.8630
0.05939.08160000.96730.8626
0.05939.14161000.96440.8630
0.05939.19162000.96240.8631
0.05939.25163000.96480.8633
0.05939.31164000.96730.8632
0.04159.36165000.96580.8633
0.04159.42166000.96880.8628
0.04159.48167000.96530.8632
0.04159.53168000.96580.8628
0.04159.59169000.96680.8629
0.04719.65170000.96040.8625
0.04719.7171000.96580.8621
0.04719.76172000.97310.8630
0.04719.82173000.96920.8626
0.04719.88174000.96730.8623
0.05289.93175000.96140.8620
0.05289.99176000.96970.8621

Framework versions

  • Transformers 4.21.0.dev0
  • Pytorch 1.12.1+cu113
  • Datasets 2.4.0
  • Tokenizers 0.12.1
endpoints_compatible
generated_from_trainer
gpt_neo
pytorch
text-generation
transformers

Contributors

mattf1n

3 commits

joaogante

1 commits

allenai/bhaskara

Model

should probably proofread and complete it, then remove this comment. -->

14

5 commits

2 linked in READMEs

updated Mar 6, 2023

See the code

README

output

Model description

This model is a fine-tuned version of EleutherAI/gpt-neo-2.7B on the Lila-IID-train/dev set from the Lila dataset.

Usage

Bhaskara was trained with the following format:

Question: ...

Answer: ...

Program:
```python
...
```

It will perform best if queried in this way.

Intended uses & limitations

If you use this model, please cite our work.

@INPROCEEDINGS{Mishra2022Lila,
  author = {
    Swaroop Mishra 
      and Matthew Finlayson 
      and Pan Lu 
      and Leonard Tang 
      and Sean Welleck 
      and Chitta Baral 
      and Tanmay Rajpurohit 
      and Oyvind Tafjord 
      and Ashish Sabharwal 
      and Peter Clark 
      and Ashwin Kalyan},
  title = {Lila: A Unified Benchmark for Mathematical Reasoning},
  booktitle = {Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing (EMNLP)},
  year = {2022}
}

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • distributed_type: multi-GPU
  • num_devices: 2
  • total_train_batch_size: 8
  • total_eval_batch_size: 8
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 10.0

Training results

Training LossEpochStepValidation LossAccuracy
No log0.061000.79300.8214
No log0.112000.75440.8290
No log0.173000.73580.8328
No log0.234000.71920.8357
0.81560.285000.70120.8397
0.81560.346000.69040.8419
0.81560.47000.68020.8440
0.81560.458000.66700.8465
0.81560.519000.65720.8486
0.72190.5710000.64990.8500
0.72190.6211000.64110.8522
0.72190.6812000.63430.8537
0.72190.7413000.62990.8546
0.72190.7914000.62210.8561
0.6620.8515000.61570.8574
0.6620.9116000.61380.8579
0.6620.9617000.60550.8595
0.6621.0218000.61430.8598
0.6621.0819000.61910.8599
0.57071.1420000.61180.8607
0.57071.1921000.61230.8611
0.57071.2522000.60890.8617
0.57071.3123000.60640.8619
0.57071.3624000.60790.8625
0.49231.4225000.60400.8625
0.49231.4826000.60300.8630
0.49231.5327000.60210.8636
0.49231.5928000.60010.8643
0.49231.6529000.59810.8644
0.49091.730000.59420.8648
0.49091.7631000.59180.8650
0.49091.8232000.59230.8659
0.49091.8733000.58840.8664
0.49091.9334000.58840.8663
0.49641.9935000.59030.8669
0.49642.0436000.64210.8655
0.49642.137000.64010.8651
0.49642.1638000.64110.8649
0.49642.2139000.63870.8645
0.3452.2740000.63620.8654
0.3452.3341000.63620.8654
0.3452.3842000.63620.8654
0.3452.4443000.63570.8655
0.3452.544000.63620.8656
0.34632.5545000.63770.8658
0.34632.6146000.63570.8660
0.34632.6747000.62940.8665
0.34632.7248000.63330.8665
0.34632.7849000.63620.8662
0.35082.8450000.63570.8666
0.35082.8951000.62990.8673
0.35082.9552000.63130.8668
0.35083.0153000.71880.8646
0.35083.0654000.70170.8656
0.2953.1255000.69820.8653
0.2953.1856000.70310.8655
0.2953.2357000.69920.8651
0.2953.2958000.69970.8653
0.2953.3559000.70410.8651
0.23483.4160000.70750.8649
0.23483.4661000.69920.8650
0.23483.5262000.70650.8647
0.23483.5863000.69970.8652
0.23483.6364000.70260.8651
0.24113.6965000.70460.8656
0.24113.7566000.70070.8655
0.24113.867000.70260.8651
0.24113.8668000.70310.8655
0.24113.9269000.70120.8658
0.2513.9770000.70510.8656
0.2514.0371000.76070.8650
0.2514.0972000.76320.8656
0.2514.1473000.75880.8655
0.2514.274000.75780.8651
0.17974.2675000.77100.8645
0.17974.3176000.76270.8648
0.17974.3777000.75830.8650
0.17974.4378000.76460.8649
0.17974.4879000.75980.8646
0.17844.5480000.76560.8650
0.17844.681000.76170.8648
0.17844.6582000.75730.8651
0.17844.7183000.76710.8648
0.17844.7784000.75630.8651
0.18274.8285000.76510.8649
0.18274.8886000.76370.8650
0.18274.9487000.76070.8654
0.18274.9988000.76070.8650
0.18275.0589000.81490.8646
0.1675.1190000.80810.8648
0.1675.1691000.81840.8644
0.1675.2292000.81400.8647
0.1675.2893000.81690.8644
0.1675.3394000.81200.8645
0.13715.3995000.81540.8643
0.13715.4596000.81790.8642
0.13715.5197000.81540.8643
0.13715.5698000.81200.8645
0.13715.6299000.81100.8650
0.14255.68100000.81590.8645
0.14255.73101000.81740.8646
0.14255.79102000.81590.8649
0.14255.85103000.81100.8639
0.14255.9104000.81350.8645
0.15055.96105000.81400.8642
0.15056.02106000.86280.8640
0.15056.07107000.85400.8644
0.15056.13108000.85300.8642
0.15056.19109000.85600.8647
0.10866.24110000.85550.8649
0.10866.3111000.86040.8644
0.10866.36112000.85690.8642
0.10866.41113000.85300.8639
0.10866.47114000.85890.8643
0.10766.53115000.85250.8639
0.10766.58116000.85790.8640
0.10766.64117000.85940.8640
0.10766.7118000.85990.8643
0.10766.75119000.85640.8640
0.11096.81120000.86330.8640
0.11096.87121000.85840.8638
0.11096.92122000.86470.8636
0.11096.98123000.85990.8635
0.11097.04124000.89790.8632
0.10287.09125000.89360.8635
0.10287.15126000.90430.8637
0.10287.21127000.89890.8642
0.10287.26128000.89360.8642
0.10287.32129000.89210.8641
0.07747.38130000.89550.8634
0.07747.43131000.89500.8636
0.07747.49132000.89940.8635
0.07747.55133000.89990.8635
0.07747.6134000.89360.8631
0.08527.66135000.90480.8634
0.08527.72136000.89600.8632
0.08527.78137000.90230.8635
0.08527.83138000.89840.8638
0.08527.89139000.90190.8635
0.08797.95140000.90140.8634
0.08798.0141000.91360.8630
0.08798.06142000.93120.8639
0.08798.12143000.93460.8635
0.08798.17144000.93070.8635
0.06118.23145000.94190.8641
0.06118.29146000.93310.8631
0.06118.34147000.93750.8636
0.06118.4148000.92920.8626
0.06118.46149000.94580.8637
0.0618.51150000.93360.8634
0.0618.57151000.94090.8630
0.0618.63152000.93900.8632
0.0618.68153000.93750.8628
0.0618.74154000.93650.8630
0.06468.8155000.93700.8628
0.06468.85156000.93550.8629
0.06468.91157000.93750.8632
0.06468.97158000.93900.8630
0.06469.02159000.97170.8630
0.05939.08160000.96730.8626
0.05939.14161000.96440.8630
0.05939.19162000.96240.8631
0.05939.25163000.96480.8633
0.05939.31164000.96730.8632
0.04159.36165000.96580.8633
0.04159.42166000.96880.8628
0.04159.48167000.96530.8632
0.04159.53168000.96580.8628
0.04159.59169000.96680.8629
0.04719.65170000.96040.8625
0.04719.7171000.96580.8621
0.04719.76172000.97310.8630
0.04719.82173000.96920.8626
0.04719.88174000.96730.8623
0.05289.93175000.96140.8620
0.05289.99176000.96970.8621

Framework versions

  • Transformers 4.21.0.dev0
  • Pytorch 1.12.1+cu113
  • Datasets 2.4.0
  • Tokenizers 0.12.1
endpoints_compatible
generated_from_trainer
gpt_neo
pytorch
text-generation
transformers

Contributors

mattf1n

3 commits

joaogante

1 commits