Update ofa_task.py

OFA-Sys · Jul 6, 2022 · 1d6c444 · 1d6c444
1 parent 630e193
commit 1d6c444
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tasks/ofa_task.py b/tasks/ofa_task.py
@@ -132,7 +132,7 @@ def get_batch_iterator(
         total_row_count = dataset.dataset.get_total_row_count()
         num_batches = math.ceil(math.ceil(total_row_count / num_shards) / max_sentences)
         if len(batch_sampler) < num_batches:
-            batch_sampler.append([1])
+            batch_sampler.append([])
 
         # return a reusable, sharded iterator
         epoch_iter = iterators.EpochBatchIterator(