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)