File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -29,6 +29,16 @@ Both codes uses multiprocessing for speedup and you have mention the number of p
2929You can download all the datasets, the compiled embeddings files, and the annoy object here:
3030https://drive.google.com/drive/folders/1FYxKooQ0dbOMZcJjkP2tfLfyhuo0bWuU?usp=sharing
3131
32+ To find the closest lines in the poetry dataset for our own text, put the text in input.txt and run:
33+
34+ > python text2poem_glove.py
35+
36+ for matched lines using glove and
37+
38+ > python text2poem_BERT.py <model_name>
39+
40+ for matched liens using BERT.
41+
3242Examples of these matching are provided at the end.
3343
3444-------------------------------------------------
@@ -46,8 +56,6 @@ Then, start the training by running:
4656You can download the trained weights here (weights will be added as the training completes):
4757https://drive.google.com/drive/folders/16eIUa26n-Zmz1lWgZywY5pYmgeadodwY?usp=sharing
4858
49- The code has been modified to support multi-gpu training and Pytorch Dataloaders for faster data loading
50-
5159-------------------------------------------------
5260
5361# Unsupervised Neural Machine Translation from (https://github.com/artetxem/undreamt):
You can’t perform that action at this time.
0 commit comments