checkpoint

Author: Guanheng Zhang

beginner_source/torchtext_translation_tutorial.py

@@ -35,8 +35,8 @@
 #
 # ::
 #
-#   python -m spacy download en
-#   python -m spacy download de
+#    python -m spacy download en
+#    python -m spacy download de
 
 import torchtext
 import torch
@@ -395,7 +395,7 @@ def count_parameters(model: nn.Module):
 
 
 def train(model: nn.Module,
-          iterator,
+          iterator: torch.utils.data.DataLoader,
           optimizer: optim.Optimizer,
           criterion: nn.Module,
           clip: float):
@@ -428,7 +428,7 @@ def train(model: nn.Module,
 
 
 def evaluate(model: nn.Module,
-             iterator,
+             iterator: torch.utils.data.DataLoader,
              criterion: nn.Module):
 
     model.eval()