liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-01 01:51:18 +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,
model_path=None,
optimizer="adam",
GPU="0",
GPU=0,
seed=None,
**kwargs
):