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

About training a genelized model #21

Open
frspring opened this issue Sep 6, 2021 · 2 comments
Open

About training a genelized model #21

frspring opened this issue Sep 6, 2021 · 2 comments

Comments

@frspring
Copy link

frspring commented Sep 6, 2021

In the readme I found the line
set batch_size to 1 when you are trining a genelized model, you can enlarge it when fine-tuning.
and since the batch_size is already 1 in the outer loop which loads the image, hould I set the ray size to 1 as well?

And why it's necessary to set batch size to 1?

@apchenstu
Copy link
Owner

I remember our plan-sweep warping don't support batch size>1, but it's not difficult to extend it to support large batch size, you can take a look at that function "build_volume_costvar_img"

@frspring
Copy link
Author

frspring commented Sep 7, 2021

Well I'm working on it. Thanks!

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