index 128 is out of bounds for axis 3 with size 128

[crazysal]

While training
Full traceback

Traceback (most recent call last):
  File "train_chart.py", line 265, in <module>
    train(training_dbs, validation_db, args.start_iter)
  File "train_chart.py", line 181, in train
    training, ind = sample_data(training_dbs[0], ind)
  File "/4/reps/DeepRule/sample/line.py", line 316, in sample_data
    return globals()[system_configs.sampling_function](db, k_ind, data_aug, debug)
  File "/4/reps/DeepRule/sample/line.py", line 259, in kp_detection
    key_heatmaps[b_ind, category, detection[2 * k + 1], detection[2 * k]] = 1
IndexError: index 128 is out of bounds for axis 3 with size 128