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,...