when I run the main code with my dataset, it comes to this error:
line 116, in train
targets_fast[idx] = targets[neighbor[random.randint(1,num_neighbor)]]
IndexError: index 5 is out of bounds for dimension 0 with size 5
how to select num_neighbor?is that based on batch_size ,or my dataset's label?
when I run the main code with my dataset, it comes to this error:
line 116, in train targets_fast[idx] = targets[neighbor[random.randint(1,num_neighbor)]] IndexError: index 5 is out of bounds for dimension 0 with size 5
how to select num_neighbor?is that based on batch_size ,or my dataset's label?