adding relu at the end of lstm to avoid NaN

model.py

@@ -112,6 +112,8 @@ class BidirectionalLSTM(nn.Module):
         out, (hidden_state, cell_state) = self.lstm(x, (h0, c0))
 
         result = self.fc(out)
+        
+        result = nn.ReLU()(result)
         return result