pytorch-image-models/timm/loss
Ross Wightman 3eb4a96eda Update AugMix, JSD, etc comments and references 2020-01-11 12:02:05 -08:00
..
__init__.py 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
cross_entropy.py Big re-org, working towards making pip/module as 'timm' 2019-06-19 17:20:51 -07:00
jsd.py Update AugMix, JSD, etc comments and references 2020-01-11 12:02:05 -08:00