fix
This commit is contained in:
committed by
Nicola Demo
parent
43682dd4ca
commit
055c1ba964
@@ -1,6 +1,8 @@
|
|||||||
"""Utils module"""
|
"""Utils module"""
|
||||||
from functools import reduce
|
from functools import reduce
|
||||||
|
|
||||||
|
from .label_tensor import LabelTensor
|
||||||
|
|
||||||
|
|
||||||
def number_parameters(model, aggregate=True, only_trainable=True): #TODO: check
|
def number_parameters(model, aggregate=True, only_trainable=True): #TODO: check
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user