Fredo Guan 3bd96609c8

Davit (#1) ...

Implement the davit model from https://arxiv.org/abs/2204.03645 and https://github.com/dingmyu/davit

2022-12-06 17:19:25 -08:00

..

data

Switch TFDS dataset to use INTEGER_ACCURATE jpeg decode by default

2022-12-05 10:21:34 -08:00

loss

Remove inplace operators when calculating the loss

2022-10-17 10:41:21 +02:00

models

Davit (#1)

2022-12-06 17:19:25 -08:00

optim

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

scheduler

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

utils

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

__init__.py

Add ported Tensorflow MaxVit weights. Add a few more CLIP ViT fine-tunes. Tweak some model tag names. Improve model tag name sorting. Update HF hub push config layout.

2022-12-05 10:21:34 -08:00

version.py

multi-weight branch version -> 0.8.0dev

2022-12-05 10:21:34 -08:00