diff --git a/tbcnn/config.py b/tbcnn/config.py index 1f8b3e1..558a40e 100644 --- a/tbcnn/config.py +++ b/tbcnn/config.py @@ -62,7 +62,7 @@ def initialize(clz, from_cmd=True, **kwargs): parser.add_argument('--batch_size', help='batch size', type=int, default=128) parser.add_argument('--learning_rate', help='learning rate', type=float, default=0.0002) parser.add_argument('--weight_decay', help='weight decay factor', type=float, default=0.0002) - parser.add_argument('--word_dim', help='dimension of node feature', type=int, default=120) + parser.add_argument('--word_dim', help='dimension of node feature', type=int, default=400) parser.add_argument('--conv_dim', help='dimension of conv feature detectors', type=int, default=50) parser.add_argument('--fc_dim',