for frist two epochs
trainloss inf -> 0.1 -> 0.03
validationloss inf->0.8->0.021
when this issue happens in the first 30% of epoch 3 i stop train to check the results in real images
the results was very bad compared to the previous model