search

LambdaLabsML / examples

Deep Learning Examples
MIT License
805 stars 103 forks source link

wrong number of channels #39

Closed Johann4DL closed 1 year ago

Johann4DL commented 1 year ago

When I run the code with my own data, I run into this error:

RuntimeError: Given groups=1, weight of size [128, 3, 3, 3], expected input[3, 1, 512, 512] to have 3 channels, but got 1 channels instead

First I thought, that my images might be grayscaled (not RGB) and thus have only one channel instead of 3, but this doesn't seem to be the case. Any ideas on how to solve the issue?

lvsi-qi commented 1 year ago

Do you solve it?

lvsi-qi commented 1 year ago

Could you help me? I will thank you very much!

kushinm commented 1 year ago

Also pinging this, I seem to be getting the same issue @Johann4DL Could you tell us how you resolved the issue?