Accuracy will decrease with the number of iterations and as the model size decreases [BatchNorm changed to BatchNorm2d]. <img width="1187" alt="image" src="https://github.com/user-attachments/assets/6736abf5-89cf-4c24-a786-dbcbcca66ce2">