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

How can I train with multi-input like (A1 ,A2……)->B? #1640

Open
kuhhg opened this issue Mar 28, 2024 · 1 comment
Open

How can I train with multi-input like (A1 ,A2……)->B? #1640

kuhhg opened this issue Mar 28, 2024 · 1 comment

Comments

@kuhhg
Copy link

kuhhg commented Mar 28, 2024

How can I train with multi-input like (A1 ,A2)->B? I used to try torch.cat to conbine (A1,A2,,A3,A4) but it would change the channel into 4 and get an error.
When I try to change the input_nc =1 and output_nc =4, the error told me assert(opt.input_nc == opt.output_nc).
So, how can I use more then 2 input into the train?

@prabathbr
Copy link

you need to disable identity_loss when using different input & output channel numbers as per paper

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