I’m using Inception_v3 model which returns a tuple (output) because of which I’m using output[0]. I’ve also edited the question as the training loop works fine and it’s the validation loop with throws the error.