Skip to content

Commit

Permalink
Set word_dimm to 400 as recommended in Aetf#1
Browse files Browse the repository at this point in the history
  • Loading branch information
hmstepanek committed Nov 8, 2021
1 parent 8a9c93f commit 6f362d4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tbcnn/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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',
Expand Down

0 comments on commit 6f362d4

Please sign in to comment.