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

callbacks.py

Added at_train_end logic to base pipeline (#1932)

2023-05-23 16:03:55 -07:00

optimizers.py

fix pytorch 2.0 optimizer no inf check error (#1919)

2023-05-15 16:49:55 +02:00

schedulers.py

Backward compatibility for torch 1.x and fix for windows (#1962)

2023-05-22 08:58:08 -07:00

trainer.py

pass grad scaler from trainer to pipeline (#1996)

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