search

researchmm / TTSR

[CVPR'20] TTSR: Learning Texture Transformer Network for Image Super-Resolution
MIT License
765 stars 115 forks source link

test/train #34

Closed hbw945 closed 3 years ago

hbw945 commented 3 years ago

Hello, can 'test/train' specify a gpu to run？ where to set it？And look forward to your reply！

FuzhiYang commented 3 years ago

You can add before the command. For example:

CUDA_VISIBLE_DEVICES="0" sh test.sh

hbw945 commented 3 years ago

Thank you for your reply, which perfectly solved my problem！

hbw945 commented 3 years ago

And can you tell me how long is the training time？Thank you

FuzhiYang commented 3 years ago

Training time is about one day.

hbw945 commented 3 years ago

I use my own dataset in the training, when every 10 epochs begin to verify, the error of dividing by zero will be reported, which refers to the no.148 line of the trainer. How can I solve this problem? Thank you！ image

FuzhiYang commented 3 years ago

You can check whether the path of CUFED test set is right or not. This error usually occurs when this path is set wrong.

hbw945 commented 3 years ago

Sorry, can you point out in which file I should modify the test path? Thank you again！！！！