pqt's picture
srgan: enable model to run with gpu
26d60e1