liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 00:06:58 +08:00
Code Activity

Fix lint error with Black

Browse Source
This commit is contained in:
D-X-Y
2021-03-08 08:27:58 +00:00
parent 03ef918dd8
commit e061443560
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
qlib/contrib/model/pytorch_utils.py
View File

@@ -3,6 +3,7 @@
import torch.nn as nn import torch.nn as nn
def count_parameters(models_or_parameters, unit="mb"): def count_parameters(models_or_parameters, unit="mb"):
if isinstance(models_or_parameters, nn.Module): if isinstance(models_or_parameters, nn.Module):
counts = sum(v.numel() for v in models_or_parameters.parameters()) counts = sum(v.numel() for v in models_or_parameters.parameters())