Ross Wightman
|
3e03b2bf3f
|
Fix a few more hiera API issues
|
2024-05-12 11:11:45 -07:00 |
|
Ross Wightman
|
211d18d8ac
|
Move norm & pool into Hiera ClassifierHead. Misc fixes, update features_intermediate() naming
|
2024-05-11 23:37:35 -07:00 |
|
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 |
|
Ross Wightman
|
ef147fd2fb
|
Add forward_intermediates API to Hiera for features_only=True support
|
2024-04-21 11:30:41 -07:00 |
|
Ross Wightman
|
d88bed6535
|
Bit more Hiera fiddling
|
2024-04-21 09:36:57 -07:00 |
|
Ross Wightman
|
8a54d2a930
|
WIP Hiera implementation. Fix #2083. Trying to get image size adaptation to work.
|
2024-04-20 09:47:17 -07:00 |