diff --git a/qlib/contrib/model/pytorch_nn.py b/qlib/contrib/model/pytorch_nn.py index f11dd8be3..308494443 100644 --- a/qlib/contrib/model/pytorch_nn.py +++ b/qlib/contrib/model/pytorch_nn.py @@ -116,7 +116,7 @@ class DNNModelPytorch(Model): seed, GPU, self.use_GPU, - weight_decay + weight_decay, ) )