Commit History

fix: check local TPU instances only
87fed1b

boris commited on

feat: add config
3f13951

boris commited on

feat(train): handle multi-hosts
5b533b5

boris commited on

style
a6252c9

boris commited on

feat: minor improvements
53dade7

boris commited on

fix: update model name
61c93f2

boris commited on

fix(train): update model name
b257ca8

boris commited on

feat: log num_params
1f57ad7

boris commited on

fix: adjust training script + dataloader
a96f4dc

boris commited on

feat(sweep): update config
de21250

boris commited on

style: use isort
d209547

boris commited on

feat(train): merge logged dict
baa52db

boris commited on

chore: move files
46f7469

boris commited on