Ross Wightman
|
6f9a0c8ef2
|
Merge branch 'master' of github.com:rwightman/pytorch-models
|
2019-04-01 11:07:05 -07:00 |
|
Ross Wightman
|
5cb1a35c6b
|
Fixup Resnext, remove alternate shortcut types
|
2019-04-01 11:03:37 -07:00 |
|
Ross Wightman
|
d87824bd65
|
Merge branch 'master' of github.com:rwightman/pytorch-models
|
2019-03-17 09:57:36 -07:00 |
|
Ross Wightman
|
45cde6f0c7
|
Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models
|
2019-03-11 22:17:42 -07:00 |
|
Ross Wightman
|
321435e6b4
|
Update resnext init
|
2019-03-10 14:24:53 -07:00 |
|
Ross Wightman
|
2295cf56c2
|
Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms
|
2019-03-10 14:23:16 -07:00 |
|
Ross Wightman
|
9d927a389a
|
Add adabound, random erasing
|
2019-03-01 22:03:42 -08: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
|
e0cfeb7d8e
|
Add some models, remove a model, tweak some models
|
2019-03-01 13:08:35 -08:00 |
|
Ross Wightman
|
31055466fc
|
Fixup validate/inference script args, fix senet init for better test accuracy
|
2019-02-22 14:07:50 -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 |
|
Ross Wightman
|
c57717d325
|
Fix tta train bug, improve logging
|
2019-02-02 10:17:04 -08:00 |
|
Ross Wightman
|
72b4d162a2
|
Increase training performance
|
2019-02-01 22:48:31 -08:00 |
|
Ross Wightman
|
5855b07ae0
|
Initial commit, puting some ol pieces together
|
2019-02-01 22:07:34 -08:00 |