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/utils
Ross Wightman 2f2b22d8c7
Disable nvfuser fma / opt level overrides per #1244
3 years ago
..
__init__.py Add --fuser arg to train/validate/benchmark scripts to select jit fuser type 3 years ago
agc.py
checkpoint_saver.py
clip_grad.py
cuda.py
distributed.py
jit.py Disable nvfuser fma / opt level overrides per #1244 3 years ago
log.py
metrics.py
misc.py
model.py Fix some formatting in utils/model.py 3 years ago
model_ema.py
random.py
summary.py