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
27bbc70d71
Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.
|
4 years ago |
.. |
__init__.py
|
Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.
|
4 years ago |
checkpoint_saver.py
|
Reorg of utils into separate modules
|
4 years ago |
cuda.py
|
Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.
|
4 years ago |
distributed.py
|
Reorg of utils into separate modules
|
4 years ago |
jit.py
|
Reorg of utils into separate modules
|
4 years ago |
log.py
|
Reorg of utils into separate modules
|
4 years ago |
metrics.py
|
Fix topn metric view regression on PyTorch 1.7
|
4 years ago |
misc.py
|
Reorg of utils into separate modules
|
4 years ago |
model.py
|
Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.
|
4 years ago |
model_ema.py
|
Add back old ModelEma and rename new one to ModelEmaV2 to avoid compat breaks in dependant code. Shuffle train script, add a few comments, remove DataParallel support, support experimental torchscript training.
|
4 years ago |
summary.py
|
Reorg of utils into separate modules
|
4 years ago |