Traceback (most recent call last):
File "/home3/ls/TCD/main.py", line 83, in
main()
File "/home3/ls/TCD/main.py", line 58, in main
train_i_cl.train_task(args, i, current_task, current_head, class_indexer, prefix=prefix)
File "/home3/ls/TCD/train/train_i_cl.py", line 339, in train_task
tcd.update_importance(args,model,train_loader,criterion)
File "/home3/ls/TCD/cl_methods/tcd.py", line 21, in update_importance
loss.backward()
File "/home/user/anaconda3/envs/ls/lib/python3.10/site-packages/torch/_tensor.py", line 487, in backward
torch.autograd.backward(
File "/home/user/anaconda3/envs/ls/lib/python3.10/site-packages/torch/autograd/init.py", line 197, in backward
Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass
File "/home/user/anaconda3/envs/ls/lib/python3.10/site-packages/torch/nn/modules/module.py", line 61, in call
raise RuntimeError("You are trying to call the hook of a dead Module!")
RuntimeError: You are trying to call the hook of a dead Module!