It seems that the dataset “oxford_flowers102” have been changed in the size, the batch of 64 generates a 48 batch at the 12th iteration and generates an shape exception. When i change the batch size per replica to 48 everything works as expected
1 Like