Attempt to add broken test to mimic transformers use case (#2272)
Attempt to add broken test to mimic transformers use case (#2272)

* Attempt to add broken test

* use wandb logger

* Update test_amp.py

Co-authored-by: William Falcon <e5e1cb06e66b4dca47141118074351278caaa7fe@columbia.edu>
williamFalcon/pytorch-lightning,williamFalcon/pytorch-lightning