Update README.md

art-programmer · Jun 28, 2019 · 929e7a3 · 929e7a3
1 parent 9dd1aef
commit 929e7a3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pytorch/README.md b/pytorch/README.md
@@ -1,3 +1,5 @@
+We used Torch for our experiments. Here we provide a PyTorch version which is not well-tested.
+
 ## Dependencies
 ```bash
 pip install -r requirements.txt
@@ -9,6 +11,9 @@ python train.py --restore=0
 ```
 Set *restore=1* to resume training from a checkpoint.
 
+## Pre-trained model
+We provide the pretrained checkpoint [here](https://drive.google.com/open?id=1e5c7308fdoCMRv0w-XduWqyjYPV4JWHS).
+
 ## Testing
 ```bash
 python train.py --task=test