Ross Wightman a58f9162d7 Missed __init__.py update for attention pooling layer add

2023-10-17 09:28:21 -07:00

..

data

…

layers

Missed __init__.py update for attention pooling layer add

2023-10-17 09:28:21 -07:00

loss

…

models

Change ijepa names, add pretrain cfg for reg experimentts

2023-10-17 07:16:17 -07:00

optim

import lion in __init__.py

2023-07-27 09:26:57 +08:00

scheduler

Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168

2022-10-07 10:43:04 -07:00

utils

…

__init__.py

…

version.py

Bump to dev 0.9.8 version

2023-09-27 10:27:47 -07:00