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.
pytorch-image-models/timm
Ross Wightman c06c739901
Merge branch 'master' into bits_and_tpu
3 years ago
..
bits Transforms, augmentation work for bits, add RandomErasing support for XLA (pushing into transforms), revamp of transform/preproc config, etc ongoing... 3 years ago
data Merge branch 'master' into bits_and_tpu 3 years ago
loss asymmetric_loss 4 years ago
models Merge branch 'master' into bits_and_tpu 3 years ago
optim Add non fused LAMB optimizer option 3 years ago
scheduler Merge branch 'master' into bits_and_tpu 3 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 3 years ago