提交 e7a033d5 编写于 作者: R ranqiu

Fix bugs of DSSM

上级 8666c629
import argparse import argparse
import itertools import itertools
import distutils.util
import reader import reader
import paddle.v2 as paddle import paddle.v2 as paddle
...@@ -56,12 +57,12 @@ parser.add_argument( ...@@ -56,12 +57,12 @@ parser.add_argument(
(ModelArch.CNN_MODE, ModelArch.FC_MODE, ModelArch.RNN_MODE)) (ModelArch.CNN_MODE, ModelArch.FC_MODE, ModelArch.RNN_MODE))
parser.add_argument( parser.add_argument(
'--share_network_between_source_target', '--share_network_between_source_target',
type=bool, type=distutils.util.strtobool,
default=False, default=False,
help="whether to share network parameters between source and target") help="whether to share network parameters between source and target")
parser.add_argument( parser.add_argument(
'--share_embed', '--share_embed',
type=bool, type=distutils.util.strtobool,
default=False, default=False,
help="whether to share word embedding between source and target") help="whether to share word embedding between source and target")
parser.add_argument( parser.add_argument(
......
import argparse import argparse
import distutils.util
import paddle.v2 as paddle import paddle.v2 as paddle
from network_conf import DSSM from network_conf import DSSM
...@@ -35,8 +36,8 @@ parser.add_argument( ...@@ -35,8 +36,8 @@ parser.add_argument(
'-b', '-b',
'--batch_size', '--batch_size',
type=int, type=int,
default=10, default=32,
help="size of mini-batch (default:10)") help="size of mini-batch (default:32)")
parser.add_argument( parser.add_argument(
'-p', '-p',
'--num_passes', '--num_passes',
...@@ -62,12 +63,12 @@ parser.add_argument( ...@@ -62,12 +63,12 @@ parser.add_argument(
(ModelArch.CNN_MODE, ModelArch.FC_MODE, ModelArch.RNN_MODE)) (ModelArch.CNN_MODE, ModelArch.FC_MODE, ModelArch.RNN_MODE))
parser.add_argument( parser.add_argument(
'--share_network_between_source_target', '--share_network_between_source_target',
type=bool, type=distutils.util.strtobool,
default=False, default=False,
help="whether to share network parameters between source and target") help="whether to share network parameters between source and target")
parser.add_argument( parser.add_argument(
'--share_embed', '--share_embed',
type=bool, type=distutils.util.strtobool,
default=False, default=False,
help="whether to share word embedding between source and target") help="whether to share word embedding between source and target")
parser.add_argument( parser.add_argument(
...@@ -80,7 +81,7 @@ parser.add_argument( ...@@ -80,7 +81,7 @@ parser.add_argument(
'--num_workers', type=int, default=1, help="num worker threads, default 1") '--num_workers', type=int, default=1, help="num worker threads, default 1")
parser.add_argument( parser.add_argument(
'--use_gpu', '--use_gpu',
type=bool, type=distutils.util.strtobool,
default=False, default=False,
help="whether to use GPU devices (default: False)") help="whether to use GPU devices (default: False)")
parser.add_argument( parser.add_argument(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册