Well actually by lowering the number of epochs that my training went It fixed itself on its own. Even though I don’t know what caused the first error to happen.I know what zero division is but not so sure how it came to be. After going for 25 epochs the loss went up from 0.3 to 2.5 that might be the case.