Replies: 1 comment 2 replies
-
For grayscale images you have to set in_channels=1. |
Beta Was this translation helpful? Give feedback.
2 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
Hi,
Thank you for the great repo. I've been trying to run the model with pre-trained weights for grayscale images but it keeps giving me errors, even when I set in_channels=1 in model instantiation. Is there a way to fix that?
Saber
Beta Was this translation helpful? Give feedback.
All reactions