Autoencoder models on huggingface have incorrect name

Question

Autoencoder models on huggingface have incorrect name

reesekneeland opened this issue a year ago · 3 comments

Hello,

The pretrained autoencoder for subject1 is called "epoch120.pth", which is what the code expects, however the pretrained models for subjects 2, 5, and 7 are called "last.pth", causing an error when the model is loaded in the Reconstructions script.

Answer 1 · 2023-10-11T12:38:01.000Z

Thank you! I have changed the names for subj 2 5 7 to be epoch120 instead of last.pth on huggingface

Answer 2 · 2023-10-18T10:21:27.000Z

Hey Reese,

Sorry to bother you. But I want to know have you tried to train that autoencoder through 'train_autoencoder.py'? It seems that it has missed one file named 'sd_image_var_autoenc.pth'.

Do you mind to help me out? Thanks!

Answer 3 · 2023-10-18T14:04:02.000Z

Hello, I have only been using the pretrained models so I have not tested the training script at all.