Ross Wightman
|
0387e6057e
|
Update binary cross ent impl to use thresholding as an option (convert soft targets from mixup/cutmix to 0, 1)
|
2021-09-23 15:45:39 -07:00 |
|
Ross Wightman
|
fb94350896
|
Update training script and loader factory to allow use of scheduler updates, repeat augment, and bce loss
|
2021-09-01 17:46:40 -07:00 |
|
Ross Wightman
|
ba9c1108a1
|
Add a BCE loss impl that converts dense targets to sparse /w smoothing as an alternate to CE w/ smoothing. For training experiments.
|
2021-09-01 17:39:28 -07:00 |
talrid
|
27fadaa922
|
asymmetric_loss
|
2020-10-16 17:12:28 +03:00 |
|
Ross Wightman
|
6c17d57a2c
|
Fix some attributions, add copyrights to some file docstrings
|
2020-07-27 13:44:56 -07:00 |
|
Ross Wightman
|
3eb4a96eda
|
Update AugMix, JSD, etc comments and references
|
2020-01-11 12:02:05 -08:00 |
|
Ross Wightman
|
7547119891
|
Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together
|
2020-01-05 19:58:59 -08:00 |
|
Ross Wightman
|
232ab7fb12
|
Working on an implementation of AugMix with JensenShannonDivergence loss that's compatible with my AutoAugment and RandAugment impl
|
2019-12-20 23:04:11 -08:00 |
|
Ross Wightman
|
aa4354f466
|
Big re-org, working towards making pip/module as 'timm'
|
2019-06-19 17:20:51 -07:00 |