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

Issue with scaling with X3 #27

Open
Ashu7397 opened this issue Apr 12, 2021 · 1 comment
Open

Issue with scaling with X3 #27

Ashu7397 opened this issue Apr 12, 2021 · 1 comment

Comments

@Ashu7397
Copy link

On trying to train the model for X3 scale, I am getting a runtime error:
Args: --chop --batch_size 4 --model CSNLN --patch_size 96 --save CSNLN_x2 --n_feats 128 --depth 12 --data_train DIV2K --save_models

src/model/attention.py:
yi = F.conv_transpose2d(yi, wi_center, stride=self.stride*self.scale, padding=self.scale)
RuntimeError: Given transposed=1, weight of size 121 128 9 9, expected input[1, 100, 32, 32] to have 121 channels, but got 100 channels instead

Other scales like X2 and X4 work well.

@SitthipongPed
Copy link

You need to change patch size for x3 as 144. Actually, the patch size should be 96, 144 ,192 for x2, x3, x4 respectively

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