pytorch-image-models/timm/loss
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
..
__init__.py 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
asymmetric_loss.py asymmetric_loss 2020-10-16 17:12:28 +03:00
binary_cross_entropy.py 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
cross_entropy.py Big re-org, working towards making pip/module as 'timm' 2019-06-19 17:20:51 -07:00
jsd.py Fix some attributions, add copyrights to some file docstrings 2020-07-27 13:44:56 -07:00