Ross Wightman dc1bb05e8e Punch cache_dir through model factory / builder / pretrain helpers. Improve some annotations in related code.

2024-12-06 10:25:06 -08:00

..

data

Add mini imagenet info files

2024-11-25 10:53:28 -08:00

layers

See if we can avoid some model / layer pickle issues with the aa attr in ConvNormAct

2024-12-03 08:02:55 -08:00

loss

…

models

Punch cache_dir through model factory / builder / pretrain helpers. Improve some annotations in related code.

2024-12-06 10:25:06 -08:00

optim

Add cautious mars, improve test reliability by skipping grad diff for first step

2024-12-02 11:29:02 -08:00

scheduler

…

utils

…

__init__.py

…

py.typed

…

version.py

Version 1.0.12

2024-12-03 10:34:52 -08:00