KaiyangZhou / pytorch-vsumm-reinforce

Unsupervised video summarization with deep reinforcement learning (AAAI'18)
MIT License
472 stars 150 forks source link

dist_mat = dist_mat.min(1, keepdim=True)[0] IndexError: Dimension out of range #88

Open Alicoo1907 opened 1 year ago

Alicoo1907 commented 1 year ago

İ created h5 file following the : https://github.com/TorRient/Video-Summarization-Pytorch

Alicoo1907 commented 1 year ago

İ created h5 dataset with my own videos. When i want to train there is a error for representativeness reward.

line 61, in compute_reward dist_mat = dist_mat.min(1, keepdim=True)[0] IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)