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

fix gan (#3955)

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