You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I've taken your changes into consideration. This code was written in a very old version of PyTorch. I've updated to the newest version, and provided a pip requirements.txt file.
Great work - thanks a lot for this!
When using the code on some test images, I had to adjust the following pieces of code:
In
Net.forward()
after line 135 I added the statementIn
Net.__init__()
change line 79 to prevent a dimension mismatch as follows:The text was updated successfully, but these errors were encountered: