|
|
|
@@ -354,7 +354,7 @@ if __name__ == "__main__": |
|
|
|
makedirR("/model/checkpoint/") |
|
|
|
parser = argparse.ArgumentParser(description="StyleGAN2 trainer") |
|
|
|
|
|
|
|
parser.add_argument("--path", type=str,default="/dataset/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=800000, help="total training iterations" |
|
|
|
|
