Closed tuanio closed 2 years ago
After reading the code I found out you have declared fully connected layer but haven't put it in forward. If something is wrong, please let me know.
Ah, thanks! I missed it.
After reading the code I found out you have declared fully connected layer but haven't put it in forward. If something is wrong, please let me know.