CalvinBackup/SimSwapPlus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
chenxuanhong
2022-02-13 14:20:26 +08:00
parent 9429c6d7be
commit a148db410c
11 changed files with 200 additions and 145 deletions
Download Patch File Download Diff File Expand all files Collapse all files
speed_test.py
+3 -2
View File
@@ -5,7 +5,7 @@
# Created Date: Thursday February 10th 2022
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Thursday, 10th February 2022 3:05:44 pm
# Last Modified: Sunday, 13th February 2022 3:04:07 am
# Modified By: Chen Xuanhong
# Copyright (c) 2022 Shanghai Jiao Tong University
#############################################################
@@ -18,12 +18,13 @@ import torch
if __name__ == '__main__':
script = "Generator_reduce_up"
script = "Generator_config"
class_name = "Generator"
arcface_ckpt= "arcface_ckpt/arcface_checkpoint.tar"
model_config={
"g_conv_dim": 512,
"g_kernel_size": 3,
"in_channel":16,
"res_num": 9
}