train baseline

This commit is contained in:
2023-07-02 15:00:33 +02:00
parent b387e9d64d
commit 455b5d02e7
4 changed files with 227 additions and 16 deletions
+7 -1
View File
@@ -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