liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-10 06:20:57 +08:00
Code Activity

Update pytorch_gru.py

Browse Source
This commit is contained in:
Wendi Li
2021-02-01 16:40:16 +00:00
committed by you-n-g
parent b4a92d55f8
commit decf74cbdf
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/contrib/model/pytorch_gru.py
View File

@@ -56,7 +56,7 @@ class GRU(Model):
early_stop=20, early_stop=20,
loss="mse", loss="mse",
optimizer="adam", optimizer="adam",
GPU="0", GPU=0,
seed=None, seed=None,
**kwargs **kwargs
): ):