Fix citation in README

vithursant · Mar 26, 2018 · 23ac961 · 23ac961
1 parent 541600f
commit 23ac961
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-# Magnet-Loss-PyTorch
+# MagnetLoss-PyTorch
 
 PyTorch implementation of the Magnet Loss for Deep Metric Learning, based on the following paper:
 
@@ -94,7 +94,7 @@ If you use MagnetLoss-PyTorch in a scientific publication, I would appreciate re
 Biblatex entry:
 
 ```latex
-@misc{vithu2018,
+@misc{MagnetLossPyTorch,
   author = {Thangarasa, Vithursan},
   title = {MagnetLoss-PyTorch},
   year = {2018},

diff --git a/pytorch-3p6-cuda90.yml b/pytorch-3p6-cuda90.yml
@@ -18,6 +18,7 @@ dependencies:
     - appdirs
     - gpustat
     - sklearn
-    - http://download.pytorch.org/whl/cu80/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
+    -
+http://download.pytorch.org/whl/cu80/torch-0.3.1-cp36-cp36m-linux_x86_64.whl
     - torchvision
     - visdom