mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
653cee9530
commit
b721e02fe1
Binary file not shown.
@ -974,6 +974,7 @@ def run(rank, world_size, args):
|
||||
p.data = pre_trained_model[n]
|
||||
else:
|
||||
print(n)
|
||||
exit()
|
||||
|
||||
num_param = sum([p.numel() for p in model.parameters()])
|
||||
logging.info(f"Number of model parameters: {num_param}")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user