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

eca depth wise

Browse Source
This commit is contained in:
chenxuanhong
2022-02-19 18:26:22 +08:00
parent e86270032b
commit db049166a0
7 changed files with 264 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files
speed_test.py
+5 -3
View File
@@ -5,7 +5,7 @@
# Created Date: Thursday February 10th 2022
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Thursday, 17th February 2022 2:33:30 am
# Last Modified: Saturday, 19th February 2022 6:23:43 pm
# Modified By: Chen Xuanhong
# Copyright (c) 2022 Shanghai Jiao Tong University
#############################################################
@@ -21,7 +21,8 @@ if __name__ == '__main__':
# cudnn.benchmark = True
# cudnn.enabled = True
# script = "Generator_modulation_up"
script = "Generator_modulation_depthwise_config"
script = "Generator_modulation_depthwise"
# script = "Generator_modulation_depthwise_config"
# script = "Generator_ori_config"
class_name = "Generator"
arcface_ckpt= "arcface_ckpt/arcface_checkpoint.tar"
@@ -32,7 +33,8 @@ if __name__ == '__main__':
"res_num": 9,
# "up_mode": "nearest",
"up_mode": "bilinear",
"res_mode": "depthwise"
"res_mode": "depthwise_eca",
"conv_mode": "depthwise_eca"
}
os.environ['CUDA_VISIBLE_DEVICES'] = str(0)