search

haoyuc / A2N

PyTorch code for our paper "Attention in Attention Network for Image Super-Resolution"
85 stars 18 forks source link

Problem with loading data #5

Open zyaqin opened 2 years ago

zyaqin commented 2 years ago

Thank you for sharing codes to us. While reproducing the code, I came across some minor issues that I would like to ask you for advice on, as follows: KE0_L@XDKXGJW@(6P4X1AP1 I carefully checked the code and found that the output of the data loader had only three parts, namely hr, lr and filename, and I was puzzled as to why there were four parts (lr,hr,_,idx_scale) of output in the code. Looking forward to your reply, thanks!

zyaqin commented 2 years ago

Please u provide the versions of cuda, cuddn, torch, and torchvision. Thanks

haoyuc commented 2 years ago

Please u provide the versions of cuda, cuddn, torch, and torchvision. Thanks

torch==0.4.1 torchvision==0.2.1

As for cuda and cuddn, since the ECS I used has been deleted, I'm not sure what version it is.

why there were four parts (lr,hr,_,idx_scale) of output in the code

The training part of the code is from RCAN. Actually, idx_scale is not necessary, you can use only three outputs.

haoyuc commented 2 years ago

In fact, you can simply use the training code from EDSR, it supports PyTorch > 1.0, you only need to add our model code A2N/code/model/aan.py to EDSR EDSR-PyTorch/src/model/ folder

zyaqin commented 2 years ago

Thank you for your reply.

In fact, you can simply use the training code from EDSR, it supports PyTorch > 1.0, you only need to add our model code A2N/code/model/aan.py to EDSR EDSR-PyTorch/src/model/ folder

Thank you for your reply. I will try to reproduce the code as suggested, thanks.