You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Ross Wightman
fb94350896
Update training script and loader factory to allow use of scheduler updates, repeat augment, and bce loss
|
3 years ago |
.. |
data
|
Update training script and loader factory to allow use of scheduler updates, repeat augment, and bce loss
|
3 years ago |
loss
|
Update training script and loader factory to allow use of scheduler updates, repeat augment, and bce loss
|
3 years ago |
models
|
Update HaloAttn comment
|
3 years ago |
optim
|
Add lamb/lars to optim init imports, remove stray comment
|
3 years ago |
scheduler
|
LR scheduler update:
|
3 years ago |
utils
|
Tweak accuracy topk safety. Fix #807
|
3 years ago |
__init__.py
|
Add default_cfg options for min_input_size / fixed_input_size, queries in model registry, and use for testing self-attn models
|
4 years ago |
version.py
|
Add efficientnetv2_rw_t defs w/ weights, and gc variant, as well as gcresnet26ts for experiments. Version 0.4.13
|
3 years ago |