(self, path)
Save the model in the passed `path`.
Save the model in the passed `path`.
Save the model in the passed path .
def save(self, path):
        """Save the model in the passed `path`."""
        device_backup = self._device
        self.set_device(torch.device('cpu'))
        torch.save(self, path)
        self.set_device(device_backup)
def save ( self , path ) : device_backup = self . _device self . set_device ( torch . device ( 'cpu' ) ) torch . save ( self , path ) self . set_device ( device_backup )