Ross Wightman 87939e6fab Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.

2022-09-23 16:08:59 -07:00

..

data

Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.

2022-09-23 16:08:59 -07: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

Add brute-force checkpoint remapping option

2022-09-23 16:07:03 -07:00

optim

Add Adan optimizer

2022-09-23 16:05:52 -07:00

scheduler

fix: change milestones to decay-milestones

2022-05-10 07:57:19 +09:00

utils

Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.

2022-09-23 16:08:59 -07:00

__init__.py

Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.

2022-01-25 21:54:13 -08:00

version.py

version 0.6.11

2022-09-23 13:54:56 -07:00