Skip to content

Commit dab9a0b

Browse files
authored
Update README.md
1 parent 89a271b commit dab9a0b

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

README.md

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,16 @@ Both codes uses multiprocessing for speedup and you have mention the number of p
2929
You can download all the datasets, the compiled embeddings files, and the annoy object here:
3030
https://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+
3242
Examples of these matching are provided at the end.
3343

3444
-------------------------------------------------
@@ -46,8 +56,6 @@ Then, start the training by running:
4656
You can download the trained weights here (weights will be added as the training completes):
4757
https://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):

0 commit comments

Comments
 (0)