Interesting!
There’s something funny with the visualizations they shouldn’t be that saturated. convert_image_dtype isn’t working because your input is already a float.
Fixed: Copy_of_learnable_resizer.ipynb - Google Drive 10
Also:

It’s clearer if you show the before and after for each image.
It’s easier to follow if you create get_resizer function that returns a concrete resizer model, then just use that.