Terrance WangandMatthew Tancik a8bce87ada pass grad scaler from trainer to pipeline (#1996) ...

* pass grad sclaer from trainer to pipeline

* lint

* add grad scaler to arg doc

---------

Co-authored-by: Matthew Tancik <tancik@berkeley.edu>

2023-05-25 09:27:58 -07:00

..

__init__.py

Update copyright (#1917)

2023-05-12 16:16:13 -07:00

base_pipeline.py

pass grad scaler from trainer to pipeline (#1996)

2023-05-25 09:27:58 -07:00

dynamic_batch.py

pass grad scaler from trainer to pipeline (#1996)

2023-05-25 09:27:58 -07:00