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

..

data

…

loss

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

models

Update HaloAttn comment

2021-09-01 17:14:31 -07:00

optim

Add lamb/lars to optim init imports, remove stray comment

2021-08-18 22:55:02 -07:00

scheduler

LR scheduler update:

2021-09-01 17:33:11 -07:00

utils

Tweak accuracy topk safety. Fix #807

2021-08-19 14:18:53 -07:00

__init__.py

…

version.py

…