Improve RNN

orifmilod · May 12, 2024 · 49bffbd · 49bffbd
1 parent dfec1a3
commit 49bffbd
Show file tree

Hide file tree

Showing 5 changed files with 924 additions and 145 deletions.
diff --git a/example/language_model/pytorch/name_classification/rnn.ipynb b/example/language_model/pytorch/name_classification/rnn.ipynb
diff --git a/example/language_model/pytorch/next_word_prediction/rnn.ipynb b/example/language_model/pytorch/next_word_prediction/rnn.ipynb
diff --git a/gigatorch/loss.py b/gigatorch/loss.py
@@ -2,7 +2,7 @@
 import numpy as np
 
 
-def squared_loss(ys, y_pred):
+def squared_loss(ys: Tensor, y_pred: Tensor) -> Tensor:
     return sum([(y_pred - y_target) ** 2 for y_target, y_pred in zip(ys, y_pred)])