Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Abnormal training results #19

Open
diamond0910 opened this issue Apr 21, 2023 · 2 comments
Open

Abnormal training results #19

diamond0910 opened this issue Apr 21, 2023 · 2 comments

Comments

@diamond0910
Copy link

Thank you for releasing the code. I train the code using the given celeba dataset. But I get abnormal training results.

I use the default script pix2pix3D/train_scripts/celeba_seg.sh.

The fakes0000012.jpg seems good.
fakes000000

But the fakes000080.jpg is abnormal.
fakes000080

Can you give me some advice? Thank you!

@dunbar12138
Copy link
Owner

Thanks for pointing it out.

It's possibly because there was extra space at the end of some lines in the pix2pix3D/train_scripts/celeba_seg.sh. So some of the training options are not correctly read.

Could you try the newly updated training script?

@diamond0910
Copy link
Author

Thank you for your reply. In fact, I couldn't run the original script due to formatting issues. I put them all in one row.

This is my script.

python train.py --outdir=logs --cfg=celeba --seg_weight=2 --data=./CelebAMask/train --mask_data=./CelebAMask/train_label --data_resolution=512 --render_mask=True --dis_mask=True --neural_rendering_resolution_initial=128 --gpus=4 --batch=4 --mbstd-group=1 --gamma=1 --gen_pose_cond=True --resume=./pretrained_models/ffhqrebalanced512-128.pkl --random_c_prob=0.5 --only_raw_recons=True --lambda_cross_view=1e-4 --lambda_d_semantic=0.1 --lambda_lpips=1 --wandb_log=False --snap=5 --no_eval=True

Is there any problem? Thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants