search

zhengziqiang / ForkGAN

ForkGAN: Seeing into the rainy night. ECCV 2020 (oral). A task-agnostic image translation framework that can boost multiple vision tasks in adverse weather conditions, including localization, semantic segmentation and object detection.
121 stars 16 forks source link

Doesn't run on GPU #2

Closed lyh7283230 closed 4 years ago

lyh7283230 commented 4 years ago

Nice work! Thank you for sharing your code. When I run your code, I just find it doesn't run on GPU. What's the matter? Looking forward to your reply.

HaoL20 commented 4 years ago

It should be the problem of dataset_dir , maybe you need to modify it here

lyh7283230 commented 4 years ago

It should be the problem of dataset_dir , maybe you need to modify it here

Well, I use your default setting and my own dataset. To be specific I put my dataset in the folder of datasets. But it can't just run on GPU.