Ross Wightman c6db4043cd Update forward_intermediates for hiera to have its own fwd impl w/ early stopping. Remove return_intermediates bool from forward(). Still an fx issue with None mask arg :(

2024-04-29 17:23:37 -07:00

..

data

…

layers

…

loss

…

models

Update forward_intermediates for hiera to have its own fwd impl w/ early stopping. Remove return_intermediates bool from forward(). Still an fx issue with None mask arg :(

2024-04-29 17:23:37 -07:00

optim

…

scheduler

…

utils

Fix #2127 move to ema device

2024-04-10 21:29:09 -07:00

__init__.py

…

version.py

Next release will be 1.0 :o

2024-04-11 08:55:27 -07:00