Spaces:
Running
Running
Commit History
feat(pjit): follow t5x style
7b5868f
fix(train): grads spec
00710bc
feat(train): improve pjit speed
f254058
fix(train): consider correct batch size
b7c7458
feat(train): custom start_preconditioning_step
8149924
feat(train): handle distributed_shampoo in pjit
032f623
feat: update distributed_shampoo + fix None spec
8a9e367
feat(train): distributed_shampoo with pjit
cc34d07
fix style
f044cb8
feat(train): restore opt_state efficiently
1bfc1b5
feat(model): clean way to load on cpu
12f323d
feat(train): load model on CPU
3d43591
feat(train): different rng per node
2d212d8
feat(train): no batch dimension with pjit
df1fe19
feat(train): progress on pjit
49597a2
feat(train): start pjit support
0081723
Load from wandb artifact (#121)
f69b21b
unverified
feat(train): update sweep config
bbbf7c8
Use DalleBartTokenizer. State restoration reverted to previous method:
ae983d7
Pedro Cuenca
commited on
fix(train): variable not defined
4c87adf
feat(train): cleanup args
a2bf605
refactor(train): cleanup
274ba73
feat: custom gradient accumulation
2d07559
fix: style
df01fa8
feat(train): use MultiSteps for gradient accumulation
4fa53a5
Accept changes suggested by linter.
9f522b8
Pedro Cuenca
commited on
Update help string for `model_name_or_path`.
290e443
Pedro Cuenca
commited on
Update `resume_from_checkpoint` to use `from_pretrained`.
bb3f53e
Pedro Cuenca
commited on
Load tokenizer associated to the model checkpoint, if possible.
a77c0d4
Pedro Cuenca
commited on
Use model configuration unless a specific one is supplied.
5ec61cc
Pedro Cuenca
commited on