Fredo Guan
|
81ca323751
|
Davit update formatting and fix grad checkpointing (#7)
fixed head to gap->norm->fc as per convnext, along with option for norm->gap->fc
failed tests due to clip convnext models, davit tests passed
|
2023-01-15 14:34:56 -08:00 |
Ross Wightman
|
57992509f9
|
Fix some formatting in utils/model.py
|
2021-10-23 20:35:36 -07:00 |
|
Ross Wightman
|
e5da481073
|
Small post-merge tweak for freeze/unfreeze, add to __init__ for utils
|
2021-10-06 17:00:27 -07:00 |
Alexander Soare
|
431e60c83f
|
Add acknowledgements for freeze_batch_norm inspiration
|
2021-10-06 14:28:49 +01:00 |
|
Alexander Soare
|
65c3d78b96
|
Freeze unfreeze functionality finalized. Tests added
|
2021-10-02 15:55:08 +01:00 |
|
Alexander Soare
|
0cb8ea432c
|
wip
|
2021-10-02 15:55:08 +01:00 |
Aman Arora
|
6b18061773
|
Add GIST to docstring for quick access
|
2021-03-29 15:33:31 +11:00 |
|
Aman Arora
|
92b1db9a79
|
update docstrings and add check on and
|
2021-03-29 10:04:51 +11:00 |
|
Aman Arora
|
b85be24054
|
update to work with fnmatch
|
2021-03-29 09:36:31 +11:00 |
|
Aman Arora
|
20626e8387
|
Add to extract stats for SPP
|
2021-03-27 05:40:04 +11:00 |
|
Ross Wightman
|
532e3b417d
|
Reorg of utils into separate modules
|
2020-09-07 13:58:09 -07:00 |