From d39235da36b6f030bbbe70f6141a26023dbf37fb Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Wed, 4 Jan 2023 16:10:28 +0100
Subject: [PATCH] random search uses cuda again
---
mu_map/training/random_search.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/mu_map/training/random_search.py b/mu_map/training/random_search.py
index e6b57b3..5f87aba 100644
--- a/mu_map/training/random_search.py
+++ b/mu_map/training/random_search.py
@@ -239,8 +239,7 @@ class RandomSearchCGAN(RandomSearch):
self.iterations = iterations
self.dir = "cgan_random_search"
validate_and_make_directory(self.dir)
- # self.device = torch.device("cuda")
- self.device = torch.device("cpu")
+ self.device = torch.device("cuda")
self.params = {}
self.dir_train = os.path.join(self.dir, "train_data")
--
GitLab