IndexError: index 1 is out of bounds for dimension 1 with size 1

The error is :

Traceback (most recent call last):
File "/home/yu/.jupyter/ngsv/2dtan/train_net.py", line 161, in
main()
File "/home/yu/.jupyter/ngsv/2dtan/train_net.py", line 155, in main
model = train(cfg, args.local_rank, args.distributed)
File "/home/yu/.jupyter/ngsv/2dtan/train_net.py", line 75, in train
arguments,
File "/home/yu/.jupyter/ngsv/2dtan/tan/engine/trainer.py", line 113, in do_train
device=cfg.MODEL.DEVICE,
File "/home/yu/.jupyter/ngsv/2dtan/tan/engine/inference.py", line 94, in inference
return evaluate(dataset=dataset, predictions=predictions, nms_thresh=nms_thresh)
File "/home/yu/.jupyter/ngsv/2dtan/tan/data/datasets/evaluation.py", line 45, in evaluate
candidates, scores = score2d_to_moments_scores(score2d, num_clips, duration)
File "/home/yu/.jupyter/ngsv/2dtan/tan/data/datasets/utils.py", line 20, in score2d_to_moments_scores
scores = score2d[grids[:,0], grids[:,1]]
IndexError: index 1 is out of bounds for dimension 1 with size 1

How can I fix it? I use torch==1.3.1 and cuda==10.2. What's your version? Thanks!

Start with tan/data/datasets/utils.py, especially score2d_to_moments_scores at line 20, and trace the score2d and grids shapes from tan/data/datasets/evaluation.py. Reproduce the evaluation path using the reported PyTorch setup and determine why the second dimension has size 1. Done means evaluation completes without the reported IndexError.

Written by the indexing model from the issue text.