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

MAISI load_unet func misses the load of scale_factor for finetune training #1935

Open
shengzhang90 opened this issue Feb 1, 2025 · 0 comments

Comments

@shengzhang90
Copy link

Hi,

During the training with scripts/diff_model_train.py file.

Sometimes, the calculate_scale_factor func will output the result of "scale_factor=nan", could you please set and fix the "scale_factor" parameter in certain json file?

Also, for the finetune training, could you please load the "scale_factor" in load_unet func since you have save the "scale_factor" in the save_checkpoint func. Thanks a lot.

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

1 participant