Ryan ab0a70dfff fix feature_info.reduction

2024-12-18 21:12:40 +08:00

..

data

Make sure trust_remote code only passed to HF datasets. Improve some docstrings.

2024-12-06 11:40:04 -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

fix feature_info.reduction

2024-12-18 21:12:40 +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

Add py.typed file as recommended by PEP 561

2024-11-14 11:26:00 -08:00

version.py

Back to dev version

2024-12-06 10:44:41 -08:00