Maybe do you have any guesses about what variables (or their gradients or something else) I should look for in the debug mode in PyCharm? I can not even imagine what causes this issue! Please help! Would be very grateful!