dalle-mini / seq2seq /run_seq2seq_flax.py

Commit History

feat: update default parameters
dbe8c41

boris commited on

feat: output_length considers bos and eos
8bb2236

boris commited on

feat: shared cache folder
42ce7dd

boris commited on

fix: accumulation vs lr
4d55db6

boris commited on

fix: wandb logging with sync_tensorboard
8ba598c

boris commited on

feat: lower default source length
48c07ca

boris commited on

fix: typos
5960e87

boris commited on

feat: gradient accumulation
c9e9575

boris commited on

feat: simplify loss function
9db361a

boris commited on

feat: padding mask not required
d61405b

boris commited on

fix: typo
833a2d5

boris commited on

feat: add adafactor
600ad79

boris commited on

feat: log metrics more frequently
498559f

boris commited on

* Make padding mask optional.
ecafe5e

Pedro Cuenca commited on

* Only perform validation if requested
32dc2d8

Pedro Cuenca commited on

Preprocessing: return "labels", "decoder_input_ids" and
df3c7bd

Pedro Cuenca commited on

Decoder: set eos to an unreachable value, set min_length=max_length to
a841a4c

Pedro Cuenca commited on

accept tsv files as inputs.
a104edb

Pedro Cuenca commited on

val_max_target_length set to OUTPUT_LENGTH
6c27b0d

Pedro Cuenca commited on

use pylogging to refer to logging.
d9f5a35

Pedro Cuenca commited on

fix typos and update requirements
de74f11

Pedro Cuenca commited on

feat: adjust seq2seq script for dalle
3f0364c

boris commited on

feat: add run_seq2seq_flax
46cb01f

boris commited on