|
|
@@ -342,7 +342,7 @@ if __name__ == "__main__": |
|
|
|
|
|
|
|
parser = argparse.ArgumentParser(description="StyleGAN2 trainer") |
|
|
|
|
|
|
|
parser.add_argument("--path", type=str,default="./mydatasets", help="path to the lmdb dataset") |
|
|
|
parser.add_argument("--path", type=str,default="/dataset", help="path to the lmdb dataset") |
|
|
|
parser.add_argument('--arch', type=str, default='stylegan2', help='model architectures (stylegan2 | swagan)') |
|
|
|
parser.add_argument( |
|
|
|
"--iter", type=int, default=1200000, help="total training iterations" |
|
|
|