Update pytorch_nn.py

2026-07-02 10:31:00 +08:00 · 2021-02-01 16:33:31 +00:00
parent 213f809148
commit bdc70c192a
1 changed files with 1 additions and 1 deletions
--- a/qlib/contrib/model/pytorch_nn.py
+++ b/qlib/contrib/model/pytorch_nn.py
@@ -60,7 +60,7 @@ class DNNModelPytorch(Model):
        lr_decay_steps=100,
        optimizer="gd",
        loss="mse",
-        GPU="0",
+        GPU=0,
        seed=None,
        weight_decay=0.0,
        **kwargs