Ross Wightman ba65dfe2c6 Dataset work ...

* support some torchvision datasets
* improvements to TFDS wrapper for subsplit handling (fix #942), shuffle seed
* add class-map support to train (fix #957)

2021-11-09 22:34:15 -08:00

..

data

Dataset work

2021-11-09 22:34:15 -08:00

loss

Update binary cross ent impl to use thresholding as an option (convert soft targets from mixup/cutmix to 0, 1)

2021-09-23 15:45:39 -07:00

models

Add ResNet101 and ResNet152 weights from higher aug RSB recipes. 81.93 and 82.82 top-1 at 224x224.

2021-11-02 17:59:16 -07:00

optim

…

scheduler

LR scheduler update:

2021-09-01 17:33:11 -07:00

utils

Fix some formatting in utils/model.py

2021-10-23 20:35:36 -07:00

__init__.py

…

version.py

Version bump to 0.5.0 for pending release post RSB and ATTN updates

2021-10-03 17:34:57 -07:00