Update pytorch_lstm.py

wendili-cs · you-n-g · commit ebc31b9bdba9 · 2021-02-02T14:46:49.000+08:00
diff --git a/qlib/contrib/model/pytorch_lstm.py b/qlib/contrib/model/pytorch_lstm.py
@@ -56,7 +56,7 @@ def __init__(
         early_stop=20,
         loss="mse",
         optimizer="adam",
-        GPU="0",
+        GPU=0,
         seed=None,
         **kwargs
     ):
