search

puckikk1202 / difffacesketch2023

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

training stage #3

Open OrangeZz0331 opened 11 months ago

OrangeZz0331 commented 11 months ago

I want to train my pre trained model, but I seem to be encountering some issues.After setting your origin dataset path and output path respectively,start training,butBut it seems to be stuck and hasn't generated a model yet,Can you tell me in detail how to train it？

And this is my project directory.Store facial images in /DiffFaceSketch/datasets/celeba_img_nobg，sketch iamges in /DiffFaceSketch/datasets/celeba_sketch_nobg.The pre trained model to be generated is stored in /DiffFaceSketch/zkc_premodels.

Thank you very much!!! project directory stuck

puckikk1202 commented 11 months ago

From your log, it seems that the training has already been initialized. Can you print more information in the training loop to see where it gets stuck?

OrangeZz0331 commented 11 months ago

It got stuck on this interface, no new things were generated, and the backend GPU was running continuously. stuck

puckikk1202 commented 11 months ago

Yes, the log shows the model is initialized and started to train. I think you may try to print more log information in the training loop to check whether your data was loaded correctly, or the loss was updated correctly, etc.

OrangeZz0331 commented 11 months ago

OK，thank you very much!!!

OrangeZz0331 commented 11 months ago

Can you share your sketch database for us to train models？I want to try it.Thank you very much! 

橙橙子 @.***

 

------------------ 原始邮件 ------------------ 发件人: @.***>; 发送时间: 2023年11月30日(星期四) 下午5:20 收件人: "puckikk1202/difffacesketch2023"; 抄送: "Author"; 主题: Re: [puckikk1202/difffacesketch2023] training stage (Issue #3)

Yes, the log shows the model is initialized and started to train. I think you may try to print more log information in the training loop to check whether your data was loaded correctly, or the loss was updated correctly, etc.

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

OrangeZz0331 commented 10 months ago

I'm sorry to bother you, but I really need your help！ 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~

Message ID: @.***>