Ross Wightman
|
f1cd1a5ce3
|
Cleanup CheckpointSaver, add support for increasing or decreasing metric, switch to prec1 metric in train loop
|
2019-04-07 10:22:55 -07:00 |
|
Ross Wightman
|
c0e6e5f3db
|
Add common model interface to pnasnet and xception, update factory
|
2019-04-06 13:59:15 -07:00 |
|
Ross Wightman
|
f2029dfb65
|
Add smooth loss
|
2019-04-05 20:50:26 -07:00 |
|
Ross Wightman
|
b0158a593e
|
Fix distributed train script
|
2019-04-05 20:49:58 -07:00 |
|
Ross Wightman
|
183d8e4aef
|
Xception model working
|
2019-04-05 12:09:25 -07:00 |
|
Ross Wightman
|
1e23727f2f
|
Update inference script for new loader style
|
2019-04-05 11:58:16 -07:00 |
|
Ross Wightman
|
58571e992e
|
Change block avgpool in senets to mean for performance issues with NVIDIA and AMP especially
|
2019-04-05 10:53:13 -07:00 |
|
Ross Wightman
|
5180f94c7e
|
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support
|
2019-04-05 10:53:04 -07:00 |
|
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 |