Ross Wightman 2b49ab7a36 Fix ResNetV2 pretrained classifier issue. Fixes #540

2021-04-03 11:18:12 -07:00

..

data

Merge remote-tracking branch 'origin/benchmark-fixes-vit_hybrids' into pit_and_vit_update

2021-04-01 12:15:34 -07:00

loss

…

models

Fix ResNetV2 pretrained classifier issue. Fixes #540

2021-04-03 11:18:12 -07:00

optim

Some further create_optimizer_v2 tweaks, remove some redudnant code, add back safe model str. Benchmark step times per batch.

2021-04-01 22:34:55 -07:00

scheduler

…

utils

Add GIST to docstring for quick access

2021-03-29 15:33:31 +11:00

__init__.py

…

version.py

Update benchmark script to add precision arg. Fix some downstream (DeiT) compat issues with latest changes. Bump version to 0.4.7

2021-04-01 16:40:12 -07:00