Segmentation mask prediction on different input image sizes

[marsbroshok (Alexander)]

Thanks a lot @sgugger, change learn.data.single_ds.tfmargs['size'] = None to predict one image per time worked like a charm!