Skip to content
Snippets Groups Projects
Commit 4a6d2e8e authored by Mateusz Klimaszewski's avatar Mateusz Klimaszewski
Browse files

Fix features/targets configuration.

parent 2512fd9f
Branches
No related tags found
No related merge requests found
...@@ -108,7 +108,7 @@ assert pretrained_tokens == null || pretrained_transformer_name == null: "Can't ...@@ -108,7 +108,7 @@ assert pretrained_tokens == null || pretrained_transformer_name == null: "Can't
features: features, features: features,
targets: targets, targets: targets,
# Whether data contains semantic relation field, bool # Whether data contains semantic relation field, bool
use_sem: if in_features("semrel") then true else false, use_sem: if in_targets("semrel") then true else false,
token_indexers: { token_indexers: {
token: if use_transformer then { token: if use_transformer then {
type: "pretrained_transformer_mismatched_tmp_fix", type: "pretrained_transformer_mismatched_tmp_fix",
...@@ -254,11 +254,11 @@ assert pretrained_tokens == null || pretrained_transformer_name == null: "Can't ...@@ -254,11 +254,11 @@ assert pretrained_tokens == null || pretrained_transformer_name == null: "Can't
layer_dropout_probability: 0.33, layer_dropout_probability: 0.33,
stacked_bilstm: { stacked_bilstm: {
input_size: input_size:
char_dim + projected_embedding_dim + (char_dim + projected_embedding_dim +
if in_features('xpostag') then xpostag_dim else 0 + (if in_features('xpostag') then xpostag_dim else 0) +
if in_features('lemma') then lemma_char_dim else 0 + (if in_features('lemma') then lemma_char_dim else 0) +
if in_features('upostag') then upostag_dim else 0 + (if in_features('upostag') then upostag_dim else 0) +
if in_features('feats') then feats_dim else 0, (if in_features('feats') then feats_dim else 0)),
hidden_size: hidden_size, hidden_size: hidden_size,
num_layers: num_layers, num_layers: num_layers,
recurrent_dropout_probability: 0.33, recurrent_dropout_probability: 0.33,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment