Skip to content

Commit fcc47cf

Browse files
authored
Merge pull request #35 from naist-nlp/update-deps
Fix a bug of xCOMET-lite loader
2 parents cf6a109 + bca3bf6 commit fcc47cf

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

mbrs/metrics/xcomet.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -126,12 +126,11 @@ def __init__(
126126
encoder_model=encoder_model,
127127
pretrained_model=pretrained_model,
128128
word_layer=word_layer,
129+
layer_transformation="softmax",
129130
validation_data=validation_data,
130131
word_level_training=word_level_training,
131132
hidden_sizes=hidden_sizes,
132133
load_pretrained_weights=load_pretrained_weights,
133-
*args,
134-
**kwargs,
135134
)
136135

137136

0 commit comments

Comments
 (0)