minibatch package error

[schko]

Thanks for your work in implementing video bert! I was receiving the following error when playing around with the code, training on a set of 300 videos with a batch size of 12:

Traceback (most recent call last):
File "VideoBERT/VideoBERT/I3D/minibatch_hkmeans.py", line 30, in
main()
File "VideoBERT/VideoBERT/I3D/minibatch_hkmeans.py", line 26, in main
hkmeans(root, prefix, 4, 12, batch_size, 15, save_dir, 'vecs', centroid_dir)
File "/opt/conda/envs/videobert/lib/python3.7/site-packages/hkmeans_minibatch-1.0.2-py3.7.egg/hkmeans_minibatch/hkmeans.py", line 99, in hkmeans
hkmeans_recursive(root, prefix, h, k, batch_size, epochs, save_dir, save_prefix, centroid_dir)
File "/opt/conda/envs/videobert/lib/python3.7/site-packages/hkmeans_minibatch-1.0.2-py3.7.egg/hkmeans_minibatch/hkmeans.py", line 91, in hkmeans_recursive
save_prefix.format(i), centroid_dir, cur_h=cur_h + 1)
File "/opt/conda/envs/videobert/lib/python3.7/site-packages/hkmeans_minibatch-1.0.2-py3.7.egg/hkmeans_minibatch/hkmeans.py", line 87, in hkmeans_recursive
save_sorted_vectors(centroids, labelled_data, batch_size, save_dir, save_prefix)
File "/opt/conda/envs/videobert/lib/python3.7/site-packages/hkmeans_minibatch-1.0.2-py3.7.egg/hkmeans_minibatch/hkmeans.py", line 56, in save_sorted_vectors
sorted_vecs.append(np.expand_dims(vectors[j], axis=0))
IndexError: index 12 is out of bounds for axis 0 with size 12

Is this an issue with dataset size?

[schko]

I think this is solved, it had to do with the batch size being a multiple of 20, and deleting my kmeans folder from past runs.