Hello @LeiaLi ! I'm impressed by your nice work and have tried the code. I have a small problem for Lines 76 and 77 in tasks/trainer.py, where 'Optimizer.zero_grad' is used after 'loss=model.forward()'. I wonder if 'optimizer.zero_grad' should be placed before 'model forward'.
Thank you very much! Looking forward to your reply.
Hello @LeiaLi ! I'm impressed by your nice work and have tried the code. I have a small problem for Lines 76 and 77 in tasks/trainer.py, where 'Optimizer.zero_grad' is used after 'loss=model.forward()'. I wonder if 'optimizer.zero_grad' should be placed before 'model forward'.
Thank you very much! Looking forward to your reply.