diff --git a/mu_map/training/cgan.py b/mu_map/training/cgan.py index aa93a3143405771c52b696484286cb78aa271d41..c9a6b9b5e190d36d8d320f89a1683eee2bbc76bf 100644 --- a/mu_map/training/cgan.py +++ b/mu_map/training/cgan.py @@ -304,7 +304,7 @@ if __name__ == "__main__": parser.add_argument( "--batch_size", type=int, - default=8, + default=64, help="the batch size used for training", ) parser.add_argument(