TypeError: einsum() takes 2 positional arguments but 3 were given

[outlookerr]

when I run
python qatm.py -s sample/sample1.jpg -t template --cuda

Result
import qatm_pytorch.py...
define model...
calculate score...
Traceback (most recent call last):
File "qatm.py", line 46, in
scores, w_array, h_array, thresh_list = run_multi_sample(model, dataset)
File "mod.py", line 333, in run_multi_sample
File "mod.py", line 306, in run_one_sample
File "mod.py", line 150, in call
TypeError: einsum() takes 2 positional arguments but 3 were given

I would appreciate if you give me any advice~

[JianliWei1995]

I update pytorch to the latest then it works.