-
Notifications
You must be signed in to change notification settings - Fork 18
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
關於train時發生的錯誤問題 #43
Comments
你好,这个warning应该没问题。目前主界面是不会输出训练过程的,所有的日志都保存在log目录下面。所以最好后台运行训练代码,然后通过 |
也有可能,你可以减小训练batch size以及patch size试试呢 |
您說的patch size是指yml檔中的GT_size以及LR_size嗎? |
GT size必须要是LR size * 4哈,保证4倍超分 |
应该是LR size 16太小了,建议至少40以上。 |
前面看着是正常的吗?这里应该是出现训练崩塌了,可以减小学习率缓解这个问题,具体可以参考这里 |
你好,
祝好 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
您好,當我執行train.py時,在dcls_arch.py的檔案發生錯誤訊息如下
File "/home/wu/DCLS/codes/config/DCLS/models/modules/dcls_arch.py", line 87, in forward
clear_features[:, i:i+1, :, :] = clear_feature_ch[:, :, ks:-ks, ks:-ks]
RuntimeError: expand(torch.cuda.FloatTensor{[64, 1, 64, 64, 2, 2]}, size=[64, 1, 64, 64]): the number of sizes provided (4) must be greater or equal to the number of dimensions in the tensor (6)
原始碼的部分是在這邊:
這程式碼中 clear_feature_ch = get_uperleft_denominator(feature_ch, kernel, kernel_P[:, i:i+1, :, :])會用到get_uperleft_denominator的這個function
這function如下:
其中convert_psf2otf這Function如下:
我對 clear_features、clear_feature_ch以及 clear_features 分別進行.shape發現它們分別的維度如以下:
torch.Size([64, 16, 64, 64])
torch.Size([64, 1, 106, 106])
torch.Size([64, 1, 106, 106, 2, 2])
請問這是為甚麼會有這個情況以及這個部分要怎麼去修改呢?謝謝!
The text was updated successfully, but these errors were encountered: