custom date about key_shot

[bigteeth43]

when i run the train.py

Traceback (most recent call last):
File "train.py", line 73, in
main()
File "train.py", line 62, in main
fscore = trainer(args, split, ckpt_path)
File "/home/yaoyc/DSNet/src/anchor_based/train.py", line 50, in train
gtscore, cps, n_frames, nfps, picks)
File "/home/yaoyc/DSNet/src/helpers/vsumm_helper.py", line 75, in get_keyshot_summ
assert pred.shape == picks.shape
AttributeError: 'str' object has no attribute 'shape'

the error happened,but i dont know whats wrong,i did what the writer said in readme