Commit History

feat: update model config + save optim
a30dbd3

boris commited on

Merge pull request #28 from patil-suraj/tpu-demo
e31a84f
unverified

Ritobrata Ghosh commited on

remove .ipynb
c879290

valhalla commited on

add tpu demo notebook
4b5a542

valhalla commited on

Merge pull request #27 from tmabraham/fix-forced-bos-token-on-demo
8f484d9
unverified

pcuenca commited on

fix forced bos token, also applying BART model to 8 samples now
8d4e13c

tmabraham commited on

Merge pull request #26 from tmabraham/generation-training-demo
fc8c230
unverified

boris commited on

demo for generation, including during training from wandb artifact
c48da33

tmabraham commited on

fix: model config
5aaf9df

boris commited on

Merge pull request #23 from khalidsaifullaah/main
eb591ff
unverified

pcuenca commited on

Merge pull request #24 from borisdayma/feat--log-model-frequently
648e404
unverified

boris commited on

feat: use bart-large-cnn
19d68bb

boris commited on

fix: log metadata
99a1ff5

boris commited on

fix: define function before it is used
d449092

boris commited on

fix: correct arg
283adc6

boris commited on

feat: save model frequently
754f876

boris commited on

feat: split script for small and big runs
5e244d0

boris commited on

feat: update test script
3cccb01

boris commited on

feat: bye bye tensorboard
533b494

boris commited on

feat: use bart large
bb3bfa6

boris commited on

YFCC metadata cleaning and encoding script
2c2f570

khalidsaifullaah commited on

Merge pull request #22 from borisdayma/feat-axis
a1c047b
unverified

boris commited on

fix: use correct key
b20769d

boris commited on

fix: log correct metrics
3fef9c1

boris commited on

feat: hardcode eval_steps
4c5e5a7

boris commited on

fix: eval_steps belongs to training_args
900136f

boris commited on

feat: eval_steps already exists in TrainingArguments
0a0080b

boris commited on

Merge branch 'main'
3ddf1c5

boris commited on

feat: set default x-axis
97a008e

boris commited on

feat: log everything through wandb
19070ab

boris commited on

Merge pull request #21 from borisdayma/feat-no_decay
b29bab7
unverified

boris commited on

feat: eval less often for faster training
f0a53ac

boris commited on

Merge pull request #20 from borisdayma/eval-interval
635402d
unverified

boris commited on

feat: no decay option
5a3211f

boris commited on

feat: use common wandb shared folder
7aa2f4b

boris commited on

feat: change default for quick tests
71c757b

boris commited on

feat: hardcoded datasets
e8709a6

boris commited on

Merge pull request #19 from pcuenca/main
f8b0895
unverified

boris commited on

Merge pull request #18 from borisdayma/change-bart-large-demo
395641f
unverified

boris commited on

Add eval_interval to evaluate and log every so often.
566d5f2

Pedro Cuenca commited on

Notebook to encode splitted YFCC100M files.
82fad8c

Pedro Cuenca commited on

change bart-large-cnn to bart-large in demo folder
5801f13

Ritobrata Ghosh commited on

Shift tokens in numpy because the built in shift function stalls.
835ea55

Pedro Cuenca commited on

fix: should be converted to array
945d86c

boris commited on

fix: labels array
6c1f112

boris commited on

fix: typo
678a62f

boris commited on

Merge pull request #17 from borisdayma/fix-model
357779a
unverified

boris commited on

fix: model config
0be4942

boris commited on

fix: correct decoder_input_ids and labels
19946be

boris commited on

Merge pull request #16 from borisdayma/feat-log_model
bf4da91
unverified

boris commited on