From c2fc8d442f1248231eab4b73e111665288b1e615 Mon Sep 17 00:00:00 2001 From: SsnL Date: Thu, 9 Nov 2017 16:08:30 -0500 Subject: update --- models/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/base_model.py') diff --git a/models/base_model.py b/models/base_model.py index 646a014..9b55afe 100644 --- a/models/base_model.py +++ b/models/base_model.py @@ -44,7 +44,7 @@ class BaseModel(): save_path = os.path.join(self.save_dir, save_filename) torch.save(network.cpu().state_dict(), save_path) if len(gpu_ids) and torch.cuda.is_available(): - network.cuda(device_id=gpu_ids[0]) # network.cuda(device=gpu_ids[0]) for the latest version. + network.cuda(gpu_ids[0]) # helper loading function that can be used by subclasses def load_network(self, network, network_label, epoch_label): -- cgit v1.2.3-70-g09d2