Ross Wightman
|
1bd92bca0e
|
Add fused_attn flag to HieraDet attn block
|
2024-08-16 22:57:49 -07:00 |
|
Ross Wightman
|
691bb54443
|
Larger min input size needed
|
2024-08-16 17:09:19 -07:00 |
|
Ross Wightman
|
de3a91a7a0
|
Add min_input_size of 128 for hieradet/sam2
|
2024-08-16 15:13:56 -07:00 |
|
Ross Wightman
|
0b05122cda
|
Fixing hieradet (sam2) tests
|
2024-08-16 14:33:40 -07:00 |
|
Ross Wightman
|
e035381171
|
Move padding out of windowing code for hieradet, fix torchscript typing issues, make pooling MaxPool unique instances across two modules
|
2024-08-16 13:36:33 -07:00 |
|
Ross Wightman
|
962958723c
|
More Hiera updates. Add forward_intermediates to hieradat/sam2 impl. Make both use same classifier module. Add coarse bool to intermediates.
|
2024-08-16 11:10:04 -07:00 |
|
Ross Wightman
|
f2cfb4c677
|
Add WIP HieraDet impl (SAM2 backbone support)
|
2024-08-15 17:58:15 -07:00 |