search

puckikk1202 / difffacesketch2023

This repository is the code for SGLDM website
MIT License
11 stars 1 forks source link

something wrong in train_ae.py #9

Open OrangeZz0331 opened 10 months ago

OrangeZz0331 commented 10 months ago

I want to know if I need to train “train_ae.py” before training “train_sketchdiff.py”？ What should I do when I encounter the following error while running “train_ae.py” ？ Looking forward to your reply~ image

OrangeZz0331 commented 10 months ago

I'm sorry to bother you, but I really need your help！

puckikk1202 commented 10 months ago

Hi, generally, there is no need to train the autoencoder (AE) if you are using our pretrained model. Regarding the error you encountered, I cannot determine whether it is due to bugs in our code or a mismatch with the dataset you used. You may find Issue #8 helpful if you encounter a similar problem.

OrangeZz0331 commented 10 months ago

Thank you for your reply！ I am trying to train the “pre-trained”mdels and I am using the dataset mentioned in the paper“CelebAMask-HQ”，After cleaning up the background of these photos I run“train_ae.py”and encountered the above error message.I don't know how to solve it, so I'm seeking your help QAQ Thank you again for your reply！Can you tell me how to solve it？