Skip to content

Commit

Permalink
adding Pytorch implementation link
Browse files Browse the repository at this point in the history
  • Loading branch information
danxuhk committed May 6, 2019
1 parent 98f2b4c commit 65eaa44
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions README.MD
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,10 @@ python train.py
The training supports multiple GPU speedup. You can modify the iter_size in the `./prototxt/solver.prototxt`, the batch_size in `gen_train_prototxt.py` and the gpu number in `train.py` to change the overall batch size. <br>
The # of overall batch size = # of gpus * batch_size * iter_size.

### Pytorch Implementation
A Pytorch implementation of our model can be found here:
https://github.com/dontLoveBugs/StructuredAttentionDepthEstimation_pytorch

### Citation
Please consider citing the following paper if the code is helpful in your research work:
<pre>
Expand Down

0 comments on commit 65eaa44

Please sign in to comment.