Ross Wightman
|
f2029dfb65
|
Add smooth loss
|
6 years ago |
Ross Wightman
|
b0158a593e
|
Fix distributed train script
|
6 years ago |
Ross Wightman
|
183d8e4aef
|
Xception model working
|
6 years ago |
Ross Wightman
|
1e23727f2f
|
Update inference script for new loader style
|
6 years ago |
Ross Wightman
|
58571e992e
|
Change block avgpool in senets to mean for performance issues with NVIDIA and AMP especially
|
6 years ago |
Ross Wightman
|
5180f94c7e
|
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support
|
6 years ago |
Ross Wightman
|
6f9a0c8ef2
|
Merge branch 'master' of github.com:rwightman/pytorch-models
|
6 years ago |
Ross Wightman
|
5cb1a35c6b
|
Fixup Resnext, remove alternate shortcut types
|
6 years ago |
Ross Wightman
|
d87824bd65
|
Merge branch 'master' of github.com:rwightman/pytorch-models
|
6 years ago |
Ross Wightman
|
45cde6f0c7
|
Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models
|
6 years ago |
Ross Wightman
|
321435e6b4
|
Update resnext init
|
6 years ago |
Ross Wightman
|
2295cf56c2
|
Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms
|
6 years ago |
Ross Wightman
|
9d927a389a
|
Add adabound, random erasing
|
6 years ago |
Ross Wightman
|
1577c52976
|
Resnext added, changes to bring it and seresnet in line with rest of models
|
6 years ago |
Ross Wightman
|
e0cfeb7d8e
|
Add some models, remove a model, tweak some models
|
6 years ago |
Ross Wightman
|
31055466fc
|
Fixup validate/inference script args, fix senet init for better test accuracy
|
6 years ago |
Ross Wightman
|
b1a5a71151
|
Update schedulers
|
6 years ago |
Ross Wightman
|
b5255960d9
|
Tweaking tanh scheduler, senet weight init (for BN), transform defaults
|
6 years ago |
Ross Wightman
|
48360625f2
|
Cycle limit on tanh sched
|
6 years ago |
Ross Wightman
|
824f42e75e
|
Forgot to include Tanh scheduler
|
6 years ago |
Ross Wightman
|
a336e5bff3
|
Minor updates
|
6 years ago |
Ross Wightman
|
cf0c280e1b
|
Cleanup tranforms, add custom schedulers, tweak senet34 model
|
6 years ago |
Ross Wightman
|
c57717d325
|
Fix tta train bug, improve logging
|
6 years ago |
Ross Wightman
|
72b4d162a2
|
Increase training performance
|
6 years ago |
Ross Wightman
|
5855b07ae0
|
Initial commit, puting some ol pieces together
|
6 years ago |