Skip to content
Snippets Groups Projects
Unverified Commit 8c7cd248 authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3941 Fix bert_large_boost ascend train failed issue.

Merge pull request !3941 from archer2049/master
parents 08f1293f ac93ca1a
No related branches found
No related tags found
No related merge requests found
......@@ -66,8 +66,6 @@ class BucketDatasetGenerator:
self.iter += 1
return self._package_data(data, key)
self.stage = 1
for value in self.data_bucket.values():
self.remaining_data += list(value)
return self._process_remaining_data()
def _package_data(self, data, key):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment