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:
@@ -42,7 +42,7 @@ class LocalformerModel(Model):
|
||||
optimizer="adam",
|
||||
reg=1e-3,
|
||||
n_jobs=10,
|
||||
GPU=2,
|
||||
GPU=0,
|
||||
seed=None,
|
||||
**kwargs
|
||||
):
|
||||
|
||||
Reference in New Issue
Block a user