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]
|
p.data = pre_trained_model[n]
|
||||||
else:
|
else:
|
||||||
print(n)
|
print(n)
|
||||||
|
exit()
|
||||||
|
|
||||||
num_param = sum([p.numel() for p in model.parameters()])
|
num_param = sum([p.numel() for p in model.parameters()])
|
||||||
logging.info(f"Number of model parameters: {num_param}")
|
logging.info(f"Number of model parameters: {num_param}")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user