liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-18 01:44:34 +08:00
Code Activity

Update pytorch_localformer.py

Browse Source
This commit is contained in:
Ying-Tao Luo
2021-07-16 15:05:32 +08:00
committed by you-n-g
parent 2df9b6e076
commit 35840606a8
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_localformer.py
View File

@@ -42,7 +42,7 @@ class LocalformerModel(Model):
optimizer="adam", optimizer="adam",
reg=1e-3, reg=1e-3,
n_jobs=10, n_jobs=10,
GPU=2, GPU=0,
seed=None, seed=None,
**kwargs **kwargs
): ):