edge-connect icon indicating copy to clipboard operation
edge-connect copied to clipboard

Sizes of tensors must match except in dimension 1

Open tendrillion opened this issue 2 years ago • 0 comments

When testing your own pictures with the shape [3,577,900],this code error which in line 251 in the src/models.py :

the error is :Sizes of tensors must match except in dimension 1. Expected size 557 but got size 556 for tensor number 1 in the list.

the code line 140 in src/models.py: outputs = self.generator(inputs)
the shape of inputs is :[1,3,557,900] but the shape of outputs is :[1,1,556,900]

any suggestions to solve this?

tendrillion avatar Dec 14 '22 10:12 tendrillion