TypeError: save_for_backward can only save variables, but argument 1 is of type int

[liu0527aa]

When I use the code to train, I met the problem.

Traceback (most recent call last):
File "/home/liuyh/pycharm-community-2021.2.2/plugins/python-ce/helpers/pydev/pydevd.py", line 1483, in _exec
pydev_imports.execfile(file, globals, locals) # execute the script
File "/home/liuyh/pycharm-community-2021.2.2/plugins/python-ce/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "/home/liuyh/demo/TPN-master/tools/train_recognizer.py", line 90, in
main()
File "/home/liuyh/demo/TPN-master/tools/train_recognizer.py", line 86, in main
logger=logger)
File "/home/liuyh/demo/TPN-master/mmaction/apis/train.py", line 60, in train_network
_non_dist_train(model, dataset, cfg, validate=validate)
File "/home/liuyh/demo/TPN-master/mmaction/apis/train.py", line 194, in _non_dist_train
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
File "/home/liuyh/miniconda3/envs/tpn/lib/python3.6/site-packages/mmcv/runner/runner.py", line 358, in run
epoch_runner(data_loaders[i], **kwargs)
File "/home/liuyh/miniconda3/envs/tpn/lib/python3.6/site-packages/mmcv/runner/runner.py", line 264, in train
self.model, data_batch, train_mode=True, **kwargs)
File "/home/liuyh/demo/TPN-master/mmaction/apis/train.py", line 37, in batch_processor
losses = model(**data)
File "/home/liuyh/miniconda3/envs/tpn/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/home/liuyh/miniconda3/envs/tpn/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 150, in forward
return self.module(*inputs[0], **kwargs[0])
File "/home/liuyh/miniconda3/envs/tpn/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/home/liuyh/demo/TPN-master/mmaction/models/recognizers/base.py", line 39, in forward
return self.forward_train(num_modalities, img_meta, **kwargs)
File "/home/liuyh/demo/TPN-master/mmaction/models/recognizers/TSN2D.py", line 115, in forward_train
x = self.segmental_consensus(x)
File "/home/liuyh/miniconda3/envs/tpn/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/home/liuyh/demo/TPN-master/mmaction/models/tenons/segmental_consensuses/simple_consensus.py", line 62, in forward
self.consensus_type)
TypeError: save_for_backward can only save variables, but argument 1 is of type int
python-BaseException

Process finished with exit code 1

Can you give me some suggestions?