Merge pull request #251 from pmeier/pytorch-cpu

increment loading index regardless of CUDA
libffcv · Feb 27, 2023 · e32280b · e32280b
2 parents 580cc95 + e5e95fd
commit e32280b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ffcv/loader/epoch_iterator.py b/ffcv/loader/epoch_iterator.py
@@ -98,7 +98,7 @@ def run(self):
                     event = ch.cuda.Event()
                     event.record(ch.cuda.default_stream())
                     events[just_finished_slot] = event
-                    b_ix += 1
+                b_ix += 1
 
         except StopIteration:
             self.output_queue.put(None)