pytorch-image-models/timm/utils
Ross Wightman a08b57e801 Fix distributed flag bug w/ flex device handling 2024-02-03 16:26:15 -08:00
..
__init__.py ModelEmaV3 and MESA experiments 2024-02-02 09:45:04 -08:00
agc.py
checkpoint_saver.py
clip_grad.py
cuda.py
decay_batch.py
distributed.py Fix distributed flag bug w/ flex device handling 2024-02-03 16:26:15 -08:00
jit.py
log.py
metrics.py
misc.py
model.py
model_ema.py forward() pass through for ema model, flag for ema warmup, comment about warmup 2024-02-03 16:24:45 -08:00
onnx.py
random.py
summary.py