Ross Wightman
|
022ed001f3
|
Update DenseNet to latest in Torchvision (torchscript compat, checkpointing, proper init). Start adding ehanced configurability, stem options...
|
2020-05-07 09:57:48 -07:00 |
|
Vyacheslav Shults
|
a7ebe09029
|
Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.
Make small code refactoring
|
2020-05-06 09:54:03 +03:00 |
|
Ross Wightman
|
13746a33fc
|
Big move, layer modules and fn to timm/models/layers
|
2020-02-09 13:13:08 -08:00 |
|
Ross Wightman
|
3bef524f9c
|
Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
|
2019-11-29 17:56:36 -08:00 |
|
Ross Wightman
|
949b7a81c4
|
Fix typo in Densenet default resolutions
|
2019-07-03 22:11:26 -07:00 |
|
Ross Wightman
|
171c0b88b6
|
Add model registry and model listing fns, refactor model_factory/create_model fn
|
2019-06-23 18:22:16 -07:00 |
|
Ross Wightman
|
6cc214bd7a
|
Consistency in model entrypoints
* move pretrained entrypoint arg to first pos to be closer to torchvision/hub
* change DPN weight URLS to my github location
|
2019-06-20 23:29:44 -07:00 |
|
Ross Wightman
|
6fc886acaf
|
Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
|
2019-06-20 17:29:25 -07:00 |
|
Ross Wightman
|
aa4354f466
|
Big re-org, working towards making pip/module as 'timm'
|
2019-06-19 17:20:51 -07:00 |
|