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 e520553e3e
Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch
2 years ago
..
__init__.py
agc.py
checkpoint_saver.py
clip_grad.py
cuda.py
decay_batch.py
distributed.py
jit.py
log.py
metrics.py
misc.py
model.py Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch 2 years ago
model_ema.py
random.py
summary.py