liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-13 15:56:57 +08:00
Code Activity

Update pytorch_nn.py

Browse Source
This commit is contained in:
Wendi Li
2021-02-01 16:33:31 +00:00
committed by you-n-g
parent 213f809148
commit bdc70c192a
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_nn.py
View File

@@ -60,7 +60,7 @@ class DNNModelPytorch(Model):
lr_decay_steps=100, lr_decay_steps=100,
optimizer="gd", optimizer="gd",
loss="mse", loss="mse",
GPU="0", GPU=0,
seed=None, seed=None,
weight_decay=0.0, weight_decay=0.0,
**kwargs **kwargs