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

Optional img2img input for diffusion reconstruction #4

Merged
merged 1 commit into from
Feb 24, 2023
Merged

Conversation

PaulScotti
Copy link
Collaborator

@PaulScotti PaulScotti commented Feb 24, 2023

updated my jupyter notebook with jimgoo’s latest contributions, also fixed problem with wandb executing for each gpu instead of just the master gpu (but only for my notebook, not jimgoo's _combo.py script)

utils.reconstruct_from_clip() is an alternative to utils.sample_images() that can take img2img reference input, as used in the Brain_to_Image scripts.

grid of images now has white background and legible plot titles

I kept the train_combined.py unchanged and didnt mess with it so as to preserve others workflows, although it would still need to be adapted to handle the new img2img feature

added rng seed to data loaders

allow user to specify num_train in get_dataloaders() via num_samples argument (useful if wanting to not use the full dataset, for testing purposes)

separated jimgoo’s main-conda.slurm from my main-singularity.slurm

@PaulScotti PaulScotti merged commit 8770138 into main Feb 24, 2023
@PaulScotti PaulScotti deleted the img_2_img branch March 1, 2023 19:10
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

Successfully merging this pull request may close these issues.

1 participant