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
76de984a5f
Fix some bugs with XLA support, logger, add hacky xla dist launch script since torch.dist.launch doesn't work
|
4 years ago |
| .. |
|
bits
|
Fix some bugs with XLA support, logger, add hacky xla dist launch script since torch.dist.launch doesn't work
|
4 years ago |
|
data
|
Fix some bugs with XLA support, logger, add hacky xla dist launch script since torch.dist.launch doesn't work
|
4 years ago |
|
loss
|
asymmetric_loss
|
4 years ago |
|
metrics
|
Fix some bugs with XLA support, logger, add hacky xla dist launch script since torch.dist.launch doesn't work
|
4 years ago |
|
models
|
Fix torchscript compat and features_only behaviour in GhostNet PR. A few minor formatting changes. Reuse existing layers.
|
4 years ago |
|
optim
|
Fix adabelief imports, remove prints, preserve memory format is the default arg for zeros_like
|
4 years ago |
|
scheduler
|
First timm.bits commit, add initial abstractions, WIP updates to train, val... some of it working
|
4 years ago |
|
utils
|
Merge pull request #550 from amaarora/wandb
|
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
|
Update README.md with latest, bump version to 0.4.8
|
4 years ago |
