From 9d9596929cdbcae785dd7dce5aeccd650477a633 Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Thu, 22 Apr 2021 11:16:43 +0200
Subject: [PATCH] Increase batch size, remove ignore classes option.
---
combo/config.multitask.template.jsonnet | 2 +-
combo/models/model.py | 3 +--
2 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/combo/config.multitask.template.jsonnet b/combo/config.multitask.template.jsonnet
index cf91ef1..3553bc1 100644
--- a/combo/config.multitask.template.jsonnet
+++ b/combo/config.multitask.template.jsonnet
@@ -199,7 +199,7 @@ assert pretrained_tokens == null || pretrained_transformer_name == null: "Can't
data_loader: {
type: "multitask",
scheduler: {
- batch_size: 10
+ batch_size: 20
},
shuffle: true,
// batch_sampler: {
diff --git a/combo/models/model.py b/combo/models/model.py
index 883934e..dd3629a 100644
--- a/combo/models/model.py
+++ b/combo/models/model.py
@@ -38,8 +38,7 @@ class NERModel(heads.Head):
super().__init__(vocab)
self.feedforward_predictor = feedforward_predictor
self._accuracy_metric = allen_metrics.CategoricalAccuracy()
- self._f1_metric = allen_metrics.SpanBasedF1Measure(vocab, tag_namespace="ner_labels", label_encoding="IOB1",
- ignore_classes=["_"])
+ self._f1_metric = allen_metrics.SpanBasedF1Measure(vocab, tag_namespace="ner_labels", label_encoding="IOB1")
self._loss = 0.0
def forward(self,
--
GitLab