From 6c71e889f891c704828df45e791d72032b26fe42 Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Thu, 22 Apr 2021 10:17:16 +0200
Subject: [PATCH] Fix loss metric.
---
combo/models/model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/combo/models/model.py b/combo/models/model.py
index cb386fb..883934e 100644
--- a/combo/models/model.py
+++ b/combo/models/model.py
@@ -53,7 +53,7 @@ class NERModel(heads.Head):
)
if tags is not None:
- self._loss = output["loss"]
+ self._loss = output["loss"].cpu().item()
self._accuracy_metric(output["probability"], tags, word_mask)
self._f1_metric(output["probability"], tags, word_mask)
--
GitLab