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

distillation

Browse Source
This commit is contained in:
chenxuanhong
2022-02-23 14:36:26 +08:00
parent db049166a0
commit a0428c8c73
9 changed files with 1017 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test.py
+2 -2
View File
@@ -5,7 +5,7 @@
# Created Date: Saturday July 3rd 2021
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Saturday, 19th February 2022 11:46:06 am
# Last Modified: Sunday, 20th February 2022 4:13:22 pm
# Modified By: Chen Xuanhong
# Copyright (c) 2021 Shanghai Jiao Tong University
#############################################################
@@ -34,7 +34,7 @@ def getParameters():
help="version name for train, test, finetune")
parser.add_argument('-c', '--cuda', type=int, default=0) # >0 if it is set as -1, program will use CPU
parser.add_argument('-s', '--checkpoint_step', type=int, default=170000,
parser.add_argument('-s', '--checkpoint_step', type=int, default=250000,
help="checkpoint epoch for test phase or finetune phase")
# test