RuntimeError: cannot perform reduction function argmax on a tensor with no elements because the operation does not have an identity

[ahxmeds]

Thanks for this amazing tutorial and repo. I had been using this code to perform tumor detection on my data set. I have been getting the error above as mentioned in the title. Did anyone else get this error?

Just to mention, I have around 45000 2D images but only around 4% of them have tumors in them, and the rest 96% do not contain any tumors. So, in order to annotate these images with no tumors, I have created empty .txt files with the same name as the image. Is this correct? Or should I be using [0, 0, 0, 0, 0] in the annotation files for the images with no tumors?

Thanks in advance.