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

update

Browse Source
This commit is contained in:
chenxuanhong
2022-02-17 16:30:33 +08:00
parent d82e46f0e7
commit 913e4916d4
17 changed files with 45302 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train_multigpu.py
+7 -7
View File
@@ -5,7 +5,7 @@
# Created Date: Tuesday April 28th 2020
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Tuesday, 15th February 2022 12:06:30 pm
# Last Modified: Thursday, 17th February 2022 2:19:29 am
# Modified By: Chen Xuanhong
# Copyright (c) 2020 Shanghai Jiao Tong University
#############################################################
@@ -31,28 +31,28 @@ def getParameters():
parser = argparse.ArgumentParser()
# general settings
parser.add_argument('-v', '--version', type=str, default='oriae_modulation',
parser.add_argument('-v', '--version', type=str, default='depthwise_config0',
help="version name for train, test, finetune")
parser.add_argument('-t', '--tag', type=str, default='oriae_modulation',
parser.add_argument('-t', '--tag', type=str, default='tiny',
help="tag for current experiment")
parser.add_argument('-p', '--phase', type=str, default="train",
choices=['train', 'finetune','debug'],
help="The phase of current project")
parser.add_argument('-c', '--gpus', type=int, nargs='+', default=[0,1,2,3]) # <0 if it is set as -1, program will use CPU
parser.add_argument('-c', '--gpus', type=int, nargs='+', default=[0,1]) # <0 if it is set as -1, program will use CPU
parser.add_argument('-e', '--ckpt', type=int, default=74,
help="checkpoint epoch for test phase or finetune phase")
# training
parser.add_argument('--experiment_description', type=str,
default="验证是否是Decoder导致的发紫")
default="尝试直接训练最小规模的网络")
parser.add_argument('--train_yaml', type=str, default="train_oriae_modulation.yaml")
parser.add_argument('--train_yaml', type=str, default="train_depthwise_modulation.yaml")
# system logger
parser.add_argument('--logger', type=str,
default="tensorboard", choices=['tensorboard', 'wandb','none'], help='system logger')
default="wandb", choices=['tensorboard', 'wandb','none'], help='system logger')
# # logs (does not to be changed in most time)
# parser.add_argument('--dataloader_workers', type=int, default=6)