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 40fea63ebe
Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts
5 years ago
..
data Add --pin-mem arg to enable dataloader pin_memory (showing more benefit in some scenarios now), also add --torchscript arg to validate.py for testing models with jit.script 5 years ago
loss
models Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts 5 years ago
optim
scheduler
__init__.py
utils.py
version.py