JingyunLiang / SwinIR

SwinIR: Image Restoration Using Swin Transformer (official repository)

Home Page:https://arxiv.org/abs/2108.10257

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training problem

YangGangZhiQi opened this issue · comments

Hi,thanks for such great work!
I am reproducing your swinIR results in realSR task. When using realSR-SwinIR-L config to train realSR model, a bug came out as below. But when using realSR-SwinIR-S config, no bug at all.
I guess the bug comes from resi_connection='3conv',only this place will produce kernel shape [60,60,1,1]. This really troubles me that the code seems like right. Why it comes out this bug. I would appreciate it if anyone could help me.
I use DDP when training realSR model.
System config:
Ubuntu 16.04
cuda 11.1
pytorch 1.8

If it's just waring, it doesn't matter.

ezoic increase your site revenue