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

showing results while training ? #119

Open
galaelized opened this issue Apr 6, 2022 · 2 comments
Open

showing results while training ? #119

galaelized opened this issue Apr 6, 2022 · 2 comments

Comments

@galaelized
Copy link

how to show generator results after every epoch during training ?

this is my current configuration

 lightweight_gan \
  --data "/content/dataset/Dataset/" \
  --num-train-steps 100000 \
  --image-size 128 \
  --name GAN2DBlood5k \
  --batch-size 32 \
  --gradient-accumulate-every 5 \
  --disc-output-size 1 \
  --dual-contrast-loss \
  --attn-res-layers [] \
  --calculate_fid_every 1000\
  --greyscale \
  --amp

using --show-progress only works after training. Also it seems that there is no longer checkpoints per epoch

@iScriptLex
Copy link

--evaluate-every 1 - show generation every 1 epoch.
--num-image-tiles 14 - make the set of 14x14 images when showing generation result.
--save-every 500 - save checkpoint every 500 epochs.

@galaelized
Copy link
Author

how to plot generator loss and discriminator loss?

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