Skip to content

Commit 87bc4b5

Browse files
committed
Implement README
1 parent cbe4ec8 commit 87bc4b5

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

README.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
In order to extract the features from the corpus proto files, run:
2+
python data_generation.py
3+
4+
In order to train a model and evaluate a model, run:
5+
python train.py --model_name="lstm_gcn_to_lstm_attention" --device=cuda:0 --print_every=10000
6+
--device="cuda" --attention=True --graph=True --iterations=500000
7+
All the possible options when running a model can be seen by running:
8+
python train.py --help
9+
10+
A pretrained version of the best performing model (as a state dictionary) can be downloaded at
11+
https://drive.google.com/file/d/1fm7hGzr-tziNhUMh8duc8s4j5gWW3uKm/view?usp=sharing
12+
13+
14+

0 commit comments

Comments
 (0)