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

update involution

Browse Source
This commit is contained in:
chenxuanhong
2022-02-12 22:13:10 +08:00
parent 5271a9f3c2
commit 9429c6d7be
14 changed files with 906 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train_multigpu.py
+5 -5
View File
@@ -5,7 +5,7 @@
# Created Date: Tuesday April 28th 2020
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Tuesday, 8th February 2022 10:50:37 pm
# Last Modified: Friday, 11th February 2022 12:10:37 am
# Modified By: Chen Xuanhong
# Copyright (c) 2020 Shanghai Jiao Tong University
#############################################################
@@ -31,9 +31,9 @@ def getParameters():
parser = argparse.ArgumentParser()
# general settings
parser.add_argument('-v', '--version', type=str, default='multigpu3',
parser.add_argument('-v', '--version', type=str, default='invoup1',
help="version name for train, test, finetune")
parser.add_argument('-t', '--tag', type=str, default='multigpu',
parser.add_argument('-t', '--tag', type=str, default='invo_upsample',
help="tag for current experiment")
parser.add_argument('-p', '--phase', type=str, default="train",
@@ -46,9 +46,9 @@ def getParameters():
# training
parser.add_argument('--experiment_description', type=str,
default="测试多GPU训练")
default="使用involution作为上采样")
parser.add_argument('--train_yaml', type=str, default="train_multigpu.yaml")
parser.add_argument('--train_yaml', type=str, default="train_Invoup.yaml")
# system logger
parser.add_argument('--logger', type=str,