mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
848 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

361 Commits (76739a7589ebde1fc6b015e5f9f3e2dc8a73299e)

Author SHA1 Message Date
Ross Wightman c6b32cbe73 A number of tweaks to arguments, epoch handling, config 
* reorganize train args
* allow resolve_data_config to be used with dict args, not just arparse
* stop incrementing epoch before save, more consistent naming vs csv, etc
* update resume and start epoch handling to match above
* stop auto-incrementing epoch in scheduler
 2019-06-28 13:49:20 -07:00
Ross Wightman 9d653b68a2 Make drop_connect rate scaling match official impl. Fixes #14 2019-06-25 09:30:36 -07:00
Ross Wightman 13c19e213d Add native PyTorch EfficientNet B1 and B2 weights. Not quite where I want them, but hitting the brick wall and moving on to other projects... 2019-06-24 13:12:04 -07:00
Ross Wightman a0275cfa2f Fix arg positions in two entrypoint aliases 2019-06-24 08:25:14 -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 8512436436 Add instagram pretrained ResNeXt models from https://pytorch.org/hub/facebookresearch_WSL-Images_resnext/, update README 2019-06-23 12:29:02 -07:00
Ross Wightman 87b92c528e Some pretrianed URL changes 
* host some of Cadene's weights on github instead of .fr for speed
* add my old port of ensemble adversarial inception resnet v2
* switch to my TF port of normal inception res v2 and change FC layer back to 'classif' for compat with ens_adv
 2019-06-21 13:57:08 -07:00
Ross Wightman 63961b36a2 Missed pnasnet entrypoint 2019-06-20 23:34:20 -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