minor change to readme

digitalbrain79 · Mar 24, 2019 · 022912f · 022912f
1 parent 4332b3c
commit 022912f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.rst b/README.rst
@@ -80,7 +80,7 @@ Get started: 30 seconds to Torchreid
         pretrained=True
     )
 
-    model = model.cuda() # move model to gpu
+    model = model.cuda()
 
     optimizer = torchreid.optim.build_optimizer(
         model,
@@ -205,4 +205,4 @@ Losses
 
 Citation
 ---------
-Please link this project in your paper
+Please link this project in your paper