1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-13 15:56:57 +08:00

Update pytorch_gru_ts.py

This commit is contained in:
Wendi Li
2021-02-01 16:40:47 +00:00
committed by you-n-g
parent ebc31b9bdb
commit b4a92d55f8

View File

@@ -58,7 +58,7 @@ class GRU(Model):
loss="mse", loss="mse",
optimizer="adam", optimizer="adam",
n_jobs=10, n_jobs=10,
GPU="0", GPU=0,
seed=None, seed=None,
**kwargs **kwargs
): ):