liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 07:16:54 +08:00
Code Activity

Update pytorch_gats.py

Browse Source
This commit is contained in:
Wendi Li
2021-02-01 16:38:17 +00:00
committed by you-n-g
parent decf74cbdf
commit f3fd5e0773
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_gats.py
View File

@@ -61,7 +61,7 @@ class GATs(Model):
with_pretrain=True, with_pretrain=True,
model_path=None, model_path=None,
optimizer="adam", optimizer="adam",
GPU="0", GPU=0,
seed=None, seed=None,
**kwargs **kwargs
): ):