Browse Source

更新 'train.py'

main
mohenghui 1 month ago
parent
commit
1087c29231
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      train.py

+ 1
- 1
train.py View File

@@ -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"


Loading…
Cancel
Save