Ross Wightman 007bc39323 Some halo and bottleneck attn code cleanup, add halonet50ts weights, use optimal crop ratios

2021-10-02 15:51:42 -07:00

..

data

Add worker_seeding arg to allow selecting old vs updated data loader worker seed for (old) experiment repeatability

2021-09-28 16:37:45 -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

Some halo and bottleneck attn code cleanup, add halonet50ts weights, use optimal crop ratios

2021-10-02 15:51:42 -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

Add efficientnetv2_rw_t defs w/ weights, and gc variant, as well as gcresnet26ts for experiments. Version 0.4.13

2021-07-09 16:46:52 -07:00