Skip to content
This repository was archived by the owner on Aug 6, 2025. It is now read-only.

wandb logging in training + add some params as args#105

Open
vinyesm wants to merge 10 commits intofacebookresearch:mainfrom
vinyesm:wandb
Open

wandb logging in training + add some params as args#105
vinyesm wants to merge 10 commits intofacebookresearch:mainfrom
vinyesm:wandb

Conversation

@vinyesm
Copy link
Copy Markdown

@vinyesm vinyesm commented Dec 12, 2024

  1. added wandb logging with samples of images from current model (see wandb screenshot)
  2. added diffusion-stepsand ema-decay as optional params during training. On small experiments 0.999 ema-decay is too high.
  3. reproduced training in small experiment (overfiting one image). To reproduce, use this colab
Screenshot 2024-12-12 at 14 39 08

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Dec 12, 2024
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants