Skip to content

Commit c9b3e99

Browse files
committed
Fix typo
1 parent 25f86a4 commit c9b3e99

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

Diff for: README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ The fine-tuned model can be found in `./checkpoint/cartoon/finetune-000600.pt`.
196196
python destylize.py --model_name FINETUNED_MODEL_NAME --batch BATCH_SIZE --iter ITERATIONS DATASET_NAME
197197
```
198198
Take the cartoon dataset for example, run:
199-
> python destylize.py --model_name fintune-000600.pt --batch 1 --iter 300 cartoon
199+
> python destylize.py --model_name finetune-000600.pt --batch 1 --iter 300 cartoon
200200
201201
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/`.
202202
To speed up destylization, set `--batch` to large value like 16.

Diff for: destylize.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def __init__(self):
2525
self.parser.add_argument("style", type=str, help="target style type")
2626
self.parser.add_argument("--truncation", type=float, default=0.7, help="truncation for intrinsic style code (content)")
2727
self.parser.add_argument("--model_path", type=str, default='./checkpoint/', help="path of the saved models")
28-
self.parser.add_argument("--model_name", type=str, default='fintune-000600.pt', help="name of the saved fine-tuned model")
28+
self.parser.add_argument("--model_name", type=str, default='finetune-000600.pt', help="name of the saved fine-tuned model")
2929
self.parser.add_argument("--data_path", type=str, default='./data/', help="path of dataset")
3030
self.parser.add_argument("--iter", type=int, default=300, help="total training iterations")
3131
self.parser.add_argument("--batch", type=int, default=1, help="batch size")

Diff for: finetune_stylegan.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -272,7 +272,7 @@ def train(args, loader, generator, discriminator, g_optim, d_optim, g_ema, devic
272272
#"args": args,
273273
#"ada_aug_p": ada_aug_p,
274274
},
275-
f"%s/%s/fintune-%06d.pt"%(args.model_path, args.style, i+1),
275+
f"%s/%s/finetune-%06d.pt"%(args.model_path, args.style, i+1),
276276
)
277277

278278
if __name__ == "__main__":

0 commit comments

Comments
 (0)