Giters

MaybeShewill-CV / attentive-gan-derainnet

Unofficial tensorflow implemention of "Attentive Generative Adversarial Network for Raindrop Removal from A Single Image (CVPR 2018) " model https://maybeshewill-cv.github.io/attentive-gan-derainnet/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

train with gpu

swordzxz opened this issue · comments

commented

hello；
Which parameters should I modify so that training can be done under the gpu?
look forward to your reply
thank you very much

commented

@swordzxz If your training process was not excepted under GPU please check if your Tensorflow or your CUDA is correctly installed:)

commented

thank you for your reply
I install the tensorflow gpu, the program can train with gpu