Update pytorch_lstm.py

2026-07-03 02:50:58 +08:00 · 2021-02-01 16:40:57 +00:00
parent 56ebe9bf36
commit ebc31b9bdb
1 changed files with 1 additions and 1 deletions
--- a/qlib/contrib/model/pytorch_lstm.py
+++ b/qlib/contrib/model/pytorch_lstm.py
@@ -56,7 +56,7 @@ class LSTM(Model):
        early_stop=20,
        loss="mse",
        optimizer="adam",
-        GPU="0",
+        GPU=0,
        seed=None,
        **kwargs
    ):