Ross Wightman 2e2687aee2 Cleanup original adafactor impl, add row/col dim heuristic that works with both conv and linear layers

2024-11-08 08:35:25 -08:00

..

data

Allow non train mode for wds reader to operate w/o sample count, exhaust iterator

2024-10-31 16:39:35 -07:00

layers

…

loss

…

models

Missed first_conv from resnet18d

2024-10-31 19:29:53 -07:00

optim

Cleanup original adafactor impl, add row/col dim heuristic that works with both conv and linear layers

2024-11-08 08:35:25 -08:00

scheduler

…

utils

…

__init__.py

…

version.py

…