Ross Wightman 214c84a235 Disable use of timm nn.Linear wrapper since AMP autocast + torchscript use appears fixed

2022-01-06 16:01:51 -08:00

..

data

Use torch.repeat_interleave() to generate repeated indices faster (#1058)

2022-01-02 14:01:06 -08:00

loss

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

models

Disable use of timm nn.Linear wrapper since AMP autocast + torchscript use appears fixed

2022-01-06 16:01:51 -08:00

optim

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

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

scheduler

Fix k_decay default arg != 1.0 in poly scheduler

2021-11-22 09:37:36 -08:00

utils

Fix some formatting in utils/model.py

2021-10-23 20:35:36 -07:00

__init__.py

…

version.py

Version bump to 0.5.0 for pending release post RSB and ATTN updates

2021-10-03 17:34:57 -07:00