RuntimeError: Given groups=1, weight of size [32, 3, 3, 3], expected input[30, 416, 416, 3] to have 3 channels, but got 416 channels instead about machine-learning-collection HOT 3 OPEN

I got something similar too, would really appreciate some assistance .

RuntimeError: Given groups=1, weight of size [256, 64, 4, 4], expected input[128, 128, 16, 16] to have 64 channels, but got 128 channels instead

Process finished with exit code 1

from machine-learning-collection.

That means your input should have 3 channels , but you give an input of 416 channels.

In your function, input size must be the form (N, C_in, H, W) and the output size is (N, C_out, H_out, W_out), where N, C, H, and W represents batch size, channel size, height and width, respectively.

I think the problem is one of two things -
First , channel mismatch in your network defining code : Take a good look at the code
Second , order of dimensions mismatch : You can permute the dimensions of your input tensor to fit the input rule (N, C_in, H, W).

from machine-learning-collection.

from machine-learning-collection.