remove duplicate

JonathanDZiegler · Sep 11, 2020 · ab21e59 · ab21e59
1 parent 4154fa9
commit ab21e59
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ctgan/synthesizer.py b/ctgan/synthesizer.py
@@ -291,7 +291,7 @@ def save(self, path):
 
         # always save a cpu model.
         device_bak = self.device
-        self.device = self.device = torch.device("cpu")
+        self.device = torch.device("cpu")
         self.generator.to(self.device)
         self.discriminator.to(self.device)