Hello everyone.
I am stuck on this problem for 2 days now.
I wrote my model and loaded data and started training but the accumulated loss is not changing ans so is the accuracy and I figured out that what is supposed to happen after optimize.step() is not actually happening.
I would be very grateful if anyone can propose any solution.