fix small bugs

mu_map/random_search/cgan.py

@@ -269,5 +269,5 @@ if __name__ == "__main__":
     args = parser.parse_args()
     device = torch.device(args.device)
 
-    random_search = cGANRandomSearch(10, torch.device("cpu"), "tmp")
+    random_search = cGANRandomSearch(args.iterations, device, args.out_dir)
     random_search.run()

mu_map/random_search/lib.py

@@ -163,7 +163,7 @@ class RandomSearch:
             self.logger.info(f"Random seed for iteration {i} is {seed}")
 
             self.setup_run()
-            # self.training.run()
+            self.training.run()
 
             score = self.eval_run()
             self.logger.info(f"Iteration {i} has score {score:.6f}")