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

GPU identification bug fix for GATs (#669)

Browse Source 
* check lexsort

* check lexsort

* lexsort comment

* lexsort comment

* fix GPU identification bug
This commit is contained in:
Mark Zhao
2021-11-03 14:41:46 +08:00
committed by GitHub
parent 7b15682c63
commit 5ee2d9496b
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
.DS_Store vendored Normal file
View File

Binary file not shown.

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

@@ -73,7 +73,7 @@ class GATs(Model):
base_model="GRU", base_model="GRU",
model_path=None, model_path=None,
optimizer="adam", optimizer="adam",
GPU="0", GPU=0,
n_jobs=10, n_jobs=10,
seed=None, seed=None,
**kwargs **kwargs