Python2 Index Error：index (4999) out of range

Question

前提・実現したいこと

Google Colaboratory上でIM2TXTのtrain.pyファイルを実行したら、以下のエラーが出てきた。それとも、最初にdata_generation.pyでデータを生成する際にループを4999以下にしないといけないでしょうか。

発生している問題・エラーメッセージ

 File "train.py", line 76, in <module>
    main(parse_args())
  File "train.py", line 65, in main
    model.train()
  File "/content/drive/IM2TXT-master/model/Model.py", line 470, in train
    trainer.run()
  File "/content/drive/IM2TXT-master/model/Model.py", line 199, in run
    self.iterate(epoch_index, captions)
  File "/content/drive/IM2TXT-master/model/Model.py", line 153, in iterate
    n_words=options['n_words'])
  File "/content/drive/IM2TXT-master/utils.py", line 49, in extract_input
    elem = features[cc[1]]
  File "/usr/local/lib/python2.7/dist-packages/scipy/sparse/csr.py", line 292, in __getitem__
    return self._get_row_slice(row, col)
  File "/usr/local/lib/python2.7/dist-packages/scipy/sparse/csr.py", line 375, in _get_row_slice
    raise IndexError('index (%d) out of range' % i)
IndexError: index (4999) out of range

試したこと

ループを4999以下にしましたが、改善されませんでした。

補足情報（FW/ツールのバージョンなど）

python2.7