Ross Wightman 80078c47bb Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.

2020-10-09 17:24:43 -07:00

..

__init__.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

checkpoint_saver.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

cuda.py

Add Adafactor and Adahessian optimizers, cleanup optimizer arg passing, add gradient clipping support.

2020-10-09 17:24:43 -07:00

distributed.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

jit.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

log.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

metrics.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

misc.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

model_ema.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

model.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00

summary.py

Reorg of utils into separate modules

2020-09-07 13:58:09 -07:00