folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Francesco Andreuzzi
2022-12-08 22:13:08 +01:00
committed by Nicola Demo
parent 43682dd4ca
commit 055c1ba964
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pina/utils.py
View File

@@ -1,6 +1,8 @@
"""Utils module"""
from functools import reduce
from .label_tensor import LabelTensor
def number_parameters(model, aggregate=True, only_trainable=True): #TODO: check
"""