CalvinBackup/neuralchen-SimSwap
1
0
Fork 0
Code Issues 288 Pull Requests 16 Actions Packages Projects Releases 2 Wiki Activity

fix save model bugs

Browse Source
This commit is contained in:
chenxuanhong
2022-04-21 20:15:21 +08:00
parent b893316e41
commit 7ed12d218f
4 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+1 -4
View File
@@ -5,7 +5,7 @@
# Created Date: Monday December 27th 2021
# Author: Chen Xuanhong
# Email: chenxuanhongzju@outlook.com
# Last Modified: Thursday, 21st April 2022 6:21:17 pm
# Last Modified: Thursday, 21st April 2022 8:10:05 pm
# Modified By: Chen Xuanhong
# Copyright (c) 2021 Shanghai Jiao Tong University
#############################################################
@@ -43,9 +43,6 @@ class TrainOptions:
self.parser.add_argument('--checkpoints_dir', type=str, default='./checkpoints', help='models are saved here')
self.parser.add_argument('--isTrain', type=str2bool, default='True')
# parser.add_argument('--use_tensorboard', type=str2bool, default='True',
# choices=['True', 'False'], help='enable the tensorboard')
# input/output sizes
self.parser.add_argument('--batchSize', type=int, default=16, help='input batch size')