train baseline
This commit is contained in:
+7
-1
@@ -42,4 +42,10 @@ class InfiniteDataLoader(DataLoader):
|
||||
# Dataset exhausted, use a new fresh iterator.
|
||||
self.dataset_iterator = super().__iter__()
|
||||
batch = next(self.dataset_iterator)
|
||||
return batch
|
||||
return batch
|
||||
|
||||
def text_to_html(lines):
|
||||
html = ''
|
||||
for line in lines:
|
||||
html += "<pre>" + line + "</pre>"
|
||||
return html
|
||||
Reference in New Issue
Block a user