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
f2e14685a8
Add force-cpu flag for train/validate, fix CPU fallback for device init, remove old force cpu flag for EMA model weights
|
4 years ago |
| .. |
|
bits
|
Add force-cpu flag for train/validate, fix CPU fallback for device init, remove old force cpu flag for EMA model weights
|
4 years ago |
|
data
|
Remove print, arg order
|
4 years ago |
|
loss
|
asymmetric_loss
|
5 years ago |
|
models
|
Merge branch 'master' into bits_and_tpu
|
4 years ago |
|
optim
|
Add lamb/lars to optim init imports, remove stray comment
|
4 years ago |
|
scheduler
|
Merge branch 'master' into bits_and_tpu
|
4 years ago |
|
utils
|
Add proper TrainState checkpoint save/load. Some reorg/refactoring and other cleanup. More to go...
|
4 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
|
4 years ago |
