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:
@@ -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
|
||||||
):
|
):
|
||||||
|
|||||||
Reference in New Issue
Block a user