Fix typo

williamyang1991 · Nov 7, 2022 · c9b3e99 · c9b3e99
1 parent 25f86a4
commit c9b3e99
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -196,7 +196,7 @@ The fine-tuned model can be found in `./checkpoint/cartoon/finetune-000600.pt`.
 python destylize.py --model_name FINETUNED_MODEL_NAME --batch BATCH_SIZE --iter ITERATIONS DATASET_NAME
 ```
 Take the cartoon dataset for example, run:
-> python destylize.py --model_name fintune-000600.pt --batch 1 --iter 300 cartoon
+> python destylize.py --model_name finetune-000600.pt --batch 1 --iter 300 cartoon
 
 The intrinsic and extrinsic style codes are saved in `./checkpoint/cartoon/instyle_code.npy` and `./checkpoint/cartoon/exstyle_code.npy`, respectively. Intermediate results are saved in `./log/cartoon/destylization/`.
 To speed up destylization, set `--batch` to large value like 16. 

diff --git a/destylize.py b/destylize.py
@@ -25,7 +25,7 @@ def __init__(self):
         self.parser.add_argument("style", type=str, help="target style type")
         self.parser.add_argument("--truncation", type=float, default=0.7, help="truncation for intrinsic style code (content)")
         self.parser.add_argument("--model_path", type=str, default='./checkpoint/', help="path of the saved models")
-        self.parser.add_argument("--model_name", type=str, default='fintune-000600.pt', help="name of the saved fine-tuned model")
+        self.parser.add_argument("--model_name", type=str, default='finetune-000600.pt', help="name of the saved fine-tuned model")
         self.parser.add_argument("--data_path", type=str, default='./data/', help="path of dataset")
         self.parser.add_argument("--iter", type=int, default=300, help="total training iterations")
         self.parser.add_argument("--batch", type=int, default=1, help="batch size")

diff --git a/finetune_stylegan.py b/finetune_stylegan.py
@@ -272,7 +272,7 @@ def train(args, loader, generator, discriminator, g_optim, d_optim, g_ema, devic
                         #"args": args,
                         #"ada_aug_p": ada_aug_p,
                     },
-                    f"%s/%s/fintune-%06d.pt"%(args.model_path, args.style, i+1),
+                    f"%s/%s/finetune-%06d.pt"%(args.model_path, args.style, i+1),
                 )
 
 if __name__ == "__main__":