Repository navigation

Commits

Commits on Mar 14, 2022
- feat: support pod (#139)
  borisdayma
  authored
  803ccbf
Commits on Feb 23, 2022
- fix: no gradient checkpointing for new model
  borisdayma
  committed
  2e02683
- feat: no gradient checkpointing for params init
  borisdayma
  committed
  b798ed3
- feat: update configs
  borisdayma
  committed
  79557f9
- feat(dev): require datasets
  borisdayma
  committed
  3d64598
- fix(train): consider schedule offset
  borisdayma
  committed
  bc4734f
- feat(train): local jax cache
  borisdayma
  committed
  9f5e879
- feat: add bucket reference to artifact
  borisdayma
  committed
  d368fb6
Commits on Feb 22, 2022
- style: lint
  borisdayma
  committed
  d5d442a
- feat: handle gradient checkpointing
  borisdayma
  committed
  5173ec7
- feat: load from bucket
  borisdayma
  committed
  1c4e839
- feat(train): save to bucket
  borisdayma
  committed
  50498e6
- feat: reduce artifact space + offset step
  borisdayma
  committed
  34cf91c
Commits on Feb 21, 2022
- feat(demo): update reference
  borisdayma
  committed
  e558000
- feat: restore weights on CPU
  borisdayma
  committed
  5f954fc
- feat(train): simplify tokenizer loading
  borisdayma
  committed
  4cb21dd
Commits on Feb 10, 2022
- doc: update README
  borisdayma
  committed
  db5a22a
- feat: cleanup notebook
  borisdayma
  committed
  5a390e8
- feat: wandb required for checkpoints
  borisdayma
  committed
  38c2c4e
- feat(demo): uncomment pip install
  borisdayma
  committed
  094e178
- feat: improve inference demo
  borisdayma
  committed
  35fe578
- fix: position embedding for generate method
  borisdayma
  committed
  ebac379
Commits on Feb 7, 2022
- feat(train): use compilation cache
  borisdayma
  committed
  da9367c
- fix: typo
  borisdayma
  committed
  68cc185
Commits on Feb 4, 2022
- fix: load from checkpoint
  borisdayma
  committed
  44b7c3e
Commits on Feb 3, 2022
- fix: style
  borisdayma
  committed
  d483294
Commits on Feb 2, 2022
- Merge branch 'main' of https://github.com/borisdayma/dalle-mini into main
  borisdayma
  committed
  0a691de
- feat: log num_parameters early
  borisdayma
  committed
  7cfe576
- fix: distributed shampoo class
  borisdayma
  committed
  696422e
- feat: update distributed_shampoo
  borisdayma
  committed
  5996680
- feat(modeling): simplify abstract_init
  borisdayma
  committed
  fa72aa7
Commits on Feb 1, 2022
- feat(train) - handle multiple nodes (#130)
  borisdayma
  authored
  0952927
- feat: handle model parallel
  borisdayma
  committed
  1bb3269
Commits on Jan 31, 2022
- feat(train): more custom x-axis
  borisdayma
  committed
  5f28cd2
Commits on Jan 30, 2022
- feat(train): split artifact into model/state (#128)
  borisdayma
  authored
  7c4c287

Branch selector

User selector

Datepicker

Commit history

Commits on Mar 14, 2022

Commits on Feb 23, 2022

Commits on Feb 22, 2022

Commits on Feb 21, 2022

Commits on Feb 10, 2022

Commits on Feb 7, 2022

Commits on Feb 4, 2022

Commits on Feb 3, 2022

Commits on Feb 2, 2022

Commits on Feb 1, 2022

Commits on Jan 31, 2022

Commits on Jan 30, 2022