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
|
5180f94c7e
|
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support
|
2019-04-05 10:53:04 -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
|
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
|
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
|
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 |