Huy Nguyen
Results
1
comments of
Huy Nguyen
I've solved the issue on lightning==2.1.3 . When rewriting any epoch_end function, if you log, just make sure that the tensor is on gpu device. If you initialize new tensor,...