1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-11 14:56:55 +08:00

Update pytorch_lstm_ts.py

This commit is contained in:
Wendi Li
2021-02-01 16:41:12 +00:00
committed by you-n-g
parent ddd68fc761
commit 56ebe9bf36

View File

@@ -58,7 +58,7 @@ class LSTM(Model):
loss="mse", loss="mse",
optimizer="adam", optimizer="adam",
n_jobs=10, n_jobs=10,
GPU="0", GPU=0,
seed=None, seed=None,
**kwargs **kwargs
): ):