The training stops after the first epoch

[AbdullahJirjees]

Hello,

I am facing this issue of training termination exacly after the first epoch is reached 100%, here is what I am getting

[ThisisKKya]

@AbdullahJirjees Since Torchsparse changed the coordinate order, you should modify the code like this.

for idx in range(invs.C[:,-1].max() + 1): 
>>>>>>
for idx in range(invs.C[:, 0].max() + 1):

cur_scene_pts = (inputs_1.C[:, -1] == idx).cpu().numpy()
cur_inv = invs.F[invs.C[:, -1] == idx].cpu().numpy()
cur_label = (all_labels.C[:, -1] == idx).cpu().numpy()
>>>>>>>>>>
cur_scene_pts = (inputs_1.C[:, 0] == idx).cpu().numpy()
cur_inv = invs.F[invs.C[:, 0] == idx].cpu().numpy()
cur_label = (all_labels.C[:, 0] == idx).cpu().numpy()