Ross Wightman fcb6258877 Add missing leaky_relu layer factory defn, update Apex/Native loss scaler interfaces to support unscaled grad clipping. Bump ver to 0.2.2 for pending release.

2020-10-02 16:19:39 -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 missing leaky_relu layer factory defn, update Apex/Native loss scaler interfaces to support unscaled grad clipping. Bump ver to 0.2.2 for pending release.

2020-10-02 16:19:39 -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