未验证 提交 2d5dd1d8 编写于 作者: L Lv Mengsi 提交者: GitHub

fix gan (#3955)

上级 a92e2129
...@@ -21,6 +21,7 @@ import sys ...@@ -21,6 +21,7 @@ import sys
import time import time
import copy import copy
import numpy as np import numpy as np
import ast
class GTrainer(): class GTrainer():
...@@ -223,7 +224,7 @@ class STGAN(object): ...@@ -223,7 +224,7 @@ class STGAN(object):
default=1024, default=1024,
help="the base fc dim in discriminator") help="the base fc dim in discriminator")
parser.add_argument( parser.add_argument(
'--use_gru', type=bool, default=True, help="whether to use GRU") '--use_gru', type=ast.literal_eval, default=True, help="whether to use GRU")
parser.add_argument( parser.add_argument(
'--lambda_cls', '--lambda_cls',
type=float, type=float,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册