try introducing dataset dir to cgan random search

mu_map/random_search/cgan.py

@@ -364,6 +364,12 @@ if __name__ == "__main__":
         default="cgan_random_search",
         help="the directory where results are stored",
     )
+    parser.add_argument(
+        "--dataset_dir",
+        type=str,
+        default="data/second",
+        help="the directory of the dataset used for training",
+    )
     parser.add_argument(
         "--iterations",
         type=int,
@@ -380,5 +386,5 @@ if __name__ == "__main__":
     args = parser.parse_args()
     device = torch.device(args.device)
 
-    random_search = cGANRandomSearch(args.iterations, device, args.out_dir)
+    random_search = cGANRandomSearch(iterations=args.iterations, device=device, out_dir=args.out_dir, dataset_dir=args.dataset_dir)
     random_search.run()