Fredo Guan 39274bd45e Update metaformers.py

2023-02-05 02:06:58 -08:00

..

data

Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn

2023-01-20 14:47:55 -08:00

layers

Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency

2023-01-20 14:44:05 -08:00

loss

…

models

Update metaformers.py

2023-02-05 02:06:58 -08:00

optim

…

scheduler

…

utils

Davit update formatting and fix grad checkpointing (#7)

2023-01-15 14:34:56 -08:00

__init__.py

…

version.py

version 0.8.8dev0

2023-01-25 18:02:48 -08:00