I've tried searching for and downloading the pretrained encoder, but it doesn't seem to work with the code you provided. I also tried other pretrained models, but they don't match the expected shape or layers. Could you update the code or provide more details to help me out?