fix problem with apex closure (#4292)
fix problem with apex closure (#4292)

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