mirror of https://github.com/coqui-ai/TTS.git
10 lines
180 B
Python
10 lines
180 B
Python
|
|
||
|
|
def get_param_size(model):
|
||
|
|
params = 0
|
||
|
|
for p in model.parameters():
|
||
|
|
tmp = 1
|
||
|
|
for x in p.size():
|
||
|
|
tmp *= x
|
||
|
|
params += tmp
|
||
|
|
return params
|