From e061443560a9cdaef837760c9a485fedd0899007 Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Mon, 8 Mar 2021 08:27:58 +0000 Subject: [PATCH] Fix lint error with Black --- qlib/contrib/model/pytorch_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/qlib/contrib/model/pytorch_utils.py b/qlib/contrib/model/pytorch_utils.py index c0d483db3..e7a8e8d67 100644 --- a/qlib/contrib/model/pytorch_utils.py +++ b/qlib/contrib/model/pytorch_utils.py @@ -3,6 +3,7 @@ import torch.nn as nn + def count_parameters(models_or_parameters, unit="mb"): if isinstance(models_or_parameters, nn.Module): counts = sum(v.numel() for v in models_or_parameters.parameters())