Unet doesnot work after dataaugmentation

You can find a small example of using the same “random” transformations here, which would be important as @bluesky314 described.