Add device comment to pytorch text-generator example

deliahu · deliahu · commit ad76dca69855 · 2019-11-04T11:33:59.000-08:00
diff --git a/examples/pytorch/text-generator/cortex.yaml b/examples/pytorch/text-generator/cortex.yaml
@@ -7,7 +7,7 @@
     path: predictor.py
     metadata:
       num_words: 20
-      device: cuda
+      device: cuda  # use "cpu" to run on CPUs
   compute:
     gpu: 1
     cpu: 1
