1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-09 22:10:56 +08:00

Update pytorch_localformer.py

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

View File

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