Hi Bhack, thanks for the references.
Answering your question. Yes, I was able to successfully train the model on the oxford-pet-iii dataset using this same code structure. What I changed was the class number, and consequently the range of labels (0 to 3 for Oxford, and 0 to 21 for Pascal VOC). I don’t understand why this doesn’t work.
I’ll study the code you sent me. Thanks.