Ross Wightman
db1fe34d0c
Update a few comment, add some references
2019-04-12 23:16:49 -07:00
Ross Wightman
9c3859fb9c
Uniform pretrained model handling.
...
* All models have 'default_cfgs' dict
* load/resume/pretrained helpers factored out
* pretrained load operates on state_dict based on default_cfg
* test all models in validate
* schedule, optim factor factored out
* test time pool wrapper applied based on default_cfg
2019-04-11 21:32:16 -07:00
Ross Wightman
1577c52976
Resnext added, changes to bring it and seresnet in line with rest of models
2019-03-01 15:44:04 -08:00
Ross Wightman
b1a5a71151
Update schedulers
2019-02-17 12:50:15 -08:00
Ross Wightman
b5255960d9
Tweaking tanh scheduler, senet weight init (for BN), transform defaults
2019-02-13 23:11:09 -08:00
Ross Wightman
48360625f2
Cycle limit on tanh sched
2019-02-08 21:47:27 -08:00
Ross Wightman
824f42e75e
Forgot to include Tanh scheduler
2019-02-08 21:14:47 -08:00
Ross Wightman
a336e5bff3
Minor updates
2019-02-08 20:56:24 -08:00
Ross Wightman
cf0c280e1b
Cleanup tranforms, add custom schedulers, tweak senet34 model
2019-02-06 20:19:11 -08:00