From ebbab31e494a7b3e56625dad5753512b0d8e11b4 Mon Sep 17 00:00:00 2001
From: Recolic K <bensl@microsoft.com>
Date: Fri, 20 Aug 2021 15:11:56 +0800
Subject: [PATCH] fix issues
---
longterm_baseline.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/longterm_baseline.py b/longterm_baseline.py
index 957d744..a1c6850 100644
--- a/longterm_baseline.py
+++ b/longterm_baseline.py
@@ -46,6 +46,7 @@ class LSTM_Shortterm_Predictor(nn.Module):
# The LSTM hidden states
# with dimensionality hidden_dim.
self.lstm = nn.LSTM(self.lstm_idim, self.lstm_odim)
+ # TODO: this model is naive. solve the bug in dataset processing.
# The linear layer that maps from hidden state space to tag space
self.out = nn.Linear(self.lstm_odim, 1)
--
GitLab