Fix validation function

Tenoke · Feb 18, 2020 · aabf91d · aabf91d
1 parent 11715bc
commit aabf91d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/reformer.py b/reformer.py
@@ -83,7 +83,7 @@ def gen_validation_inputs(n_devices):
         # different validation each time but consistent across the run
     ids = next(gen_inputs(n_devices))
     while True:
-        return ids
+        yield ids
 
 
 def create_fixed_training_schedule(lr):

diff --git a/requirements-colab.txt b/requirements-colab.txt
@@ -1,4 +1,5 @@
 jax
 jaxlib
 git+https://github.com/google/[email protected]
-tokenizers
+tokenizers
+gin