Traceback (most recent call last):
File "prune.py", line 15, in
effort = get_effort_score(args)
File "/workspace/code/DEALRec/code/prune/effort_score.py", line 154, in get_effort_score
trainer = Effort_Trainer(
File "/root/anaconda3/envs/DEALRec/lib/python3.8/site-packages/transformers/trainer.py", line 504, in init
self._move_model_to_device(model, args.device)
File "/root/anaconda3/envs/DEALRec/lib/python3.8/site-packages/transformers/trainer.py", line 728, in _move_model_to_device
model = model.to(device)
AttributeError: '_IncompatibleKeys' object has no attribute 'to'