We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cbe4ec8 commit 87bc4b5Copy full SHA for 87bc4b5
README.md
@@ -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