docstring

Author: R-Palazzo

ctgan/synthesizers/base.py

@@ -107,7 +107,7 @@ def __setstate__(self, state):
             state['random_states'] = (current_numpy_state, current_torch_state)
 
         self.__dict__ = state
-        device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
+        device = _set_device(enable_gpu=True)
         self.set_device(device)
 
     def save(self, path):

ctgan/synthesizers/ctgan.py

@@ -143,7 +143,7 @@ class CTGAN(BaseSynthesizer):
             Whether to attempt to use GPU for computation.
             Defaults to ``True``.
         cuda (bool):
-            ** Deprecated ** Whether to attempt to use cuda for GPU computation.
+            **Deprecated** Whether to attempt to use cuda for GPU computation.
             If this is False or CUDA is not available, CPU will be used.
             Defaults to ``True``.
     """