Ross Wightman f8a63a3b71 Add worker_init_fn to loader for numpy seed per worker

2021-09-23 15:44:38 -07:00

..

data

Add worker_init_fn to loader for numpy seed per worker

2021-09-23 15:44:38 -07:00

loss

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

models

Use reshape instead of view in std_conv, causing issues in recent PyTorch in channels_last

2021-09-23 15:43:48 -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

Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models

2021-04-12 11:54:22 -07:00

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