search

IBM / MAX-Image-Resolution-Enhancer

Upscale an image by a factor of 4, while generating photo-realistic details.
https://developer.ibm.com/exchanges/models/all/max-image-resolution-enhancer/
Apache License 2.0
984 stars 155 forks source link

how to use cuda ?? #45

Open 5p4r70n opened 3 years ago

5p4r70n commented 3 years ago

is that possible to use cuda for processing ....?

guanyang-nb commented 3 years ago

same question

xuhdev commented 3 years ago

I don't think it's possible to use CUDA at this stage.

guanyang-nb commented 3 years ago

It's trained on cpu?

xuhdev commented 3 years ago

This repo only performs inference. It was trained elsewhere: https://github.com/brade31919/SRGAN-tensorflow

guanyang-nb commented 3 years ago

thanks