Problem with using Fastai at prediction stage

Hi all,

I got a problem at the prediction stage. It fails to predict the rle_mask for each image, could anyone help me out with this please? I’m not sure what went wrong in my code …

https://github.com/jimmy2002916/TGS/blob/master/code/tgs.py