Closed mmcs-work closed 4 years ago
The issue is that you have the postprocess_fn in a list. It should just be a single function.
The issue is that you have the postprocess_fn in a list. It should just be a single function.
Thanks for the reply.
I am surprised that it didn't create any problem in the training phase. (Because post-process function is applied in that case as well). Or am I missing something?
Also so in case there are multiple post-process functions one wants to apply, they have to create a wrapper function that would do all those works under the hood and just pass this single wrapper function in postprocess_fn
?
I am trying to train the T5 model from scratch using the IMDB dataset. Instead of using the TensorFlow datasets I am using the raw dataset and preprocessing it accordingly. With the same dataset when I train the model it works correctly. (Loss curve is reasonable). But when I ran
model.eval()
on the test set or even the train set (the same data with which model is trained), it gets this error.Here is the notebook link.