My model works fine in pytorch But when I try to transfer to lightning,it is training slowly
In pytorch trainning 1 epochs only need 20 secs, but in lightning it cost 120 secs I check my RTX3090 usage is very low ( < 10%) Any idea?
