pytorch训练跑着好好的, 断了:
Traceback (most recent call last):
File "main_multi_model_test.py", line 147, in <module>
main()
File "main_multi_model_test.py", line 119, in main
train_loss, train_acc, train_bacc = train(model, optimizer, train_loader, criterions, taskNum,