Enable inference on CPU

amazon-science / gan-control

This package provides a pythorch implementation of "GAN-Control: Explicitly Controllable GANs", ICCV 2021.

Apache License 2.0

135 stars 23 forks source link

Closed hysts closed 2 years ago

hysts commented 2 years ago

Issue #, if available:

Description of changes:

Added the argument device and replaced .cuda() with .to(self.device) to enable inference on CPU.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.