small changes

vince62s · Oct 20, 2023 · e380b43 · e380b43
1 parent 0d289c4
commit e380b43
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -80,7 +80,7 @@ pip install -r requirements.opt.txt
 
 ## Manual installation of some dependencies
 
-Apex is highly recommended to have fast performance (especially the legacy fusedadam optimizer)
+Apex is highly recommended to have fast performance (especially the legacy fusedadam optimizer and FusedRMSNorm)
 
 ```shell
 git clone https://github.com/NVIDIA/apex

diff --git a/onmt/inputters/dynamic_iterator.py b/onmt/inputters/dynamic_iterator.py
@@ -351,7 +351,7 @@ def __iter__(self):
 
 class OnDeviceDatasetIter:
     def __init__(self, data_iter, device):
-        self.data_iter = iter(data_iter)
+        self.data_iter = data_iter
         self.device = device
 
     def __iter__(self):

diff --git a/requirements.opt.txt b/requirements.opt.txt
@@ -1,5 +1,4 @@
 pyrouge
-git+https://github.com/NVIDIA/apex.git@700d6825e205732c1d6be511306ca4e595297070
 sentencepiece>=0.1.94,<0.1.98
 subword-nmt>=0.3.7
 rapidfuzz