Hello,
I am getting the below error while trying to finetune my own data set.

Traceback (most recent call last):
File “finetune.py”, line 254, in 
main()
File “finetune.py”, line 127, in main
train(TrainImgLoader, model, optimizer, log, epoch)
File “finetune.py”, line 171, in train
for x in range(num_out)]
File “finetune.py”, line 171, in 
for x in range(num_out)]
IndexError: too many indices for tensor of dimension 3

This is the finetune.py 18 code. Please let me know how to do this