search

IIGROUP / MANIQA

[CVPRW 2022] MANIQA: Multi-dimension Attention Network for No-Reference Image Quality Assessment
Apache License 2.0
307 stars 36 forks source link

some question for dataparallel #33

Open WFLiu0327 opened 1 year ago

WFLiu0327 commented 1 year ago

when i try to train this model on double GPUs, an error occurred, suggesting that in models.maniqa.py in line 117 x = torch.cat((x6, x7, x8, x9), dim=2) appeared some data in CUDA0, some data in cuda1, please ask how this problem should be solved? I don't see some suitable solution on the internet.

TianheWu commented 1 year ago

Hi, I know this question. This reason is the problem of implementation. It just support one GPU to training.

TianheWu commented 1 year ago

One solution is to return the output of each layer directly, without using the SaveOut module