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
b721e02fe1
commit
6e27993921
Binary file not shown.
@ -971,8 +971,10 @@ def run(rank, world_size, args):
|
|||||||
|
|
||||||
for n, p in model.named_parameters():
|
for n, p in model.named_parameters():
|
||||||
if 'layer' not in n:
|
if 'layer' not in n:
|
||||||
p.data = pre_trained_model[n]
|
try: p.data = pre_trained_model[n]
|
||||||
|
except: print(f'pre-trained model has no parameterd named {n}.')
|
||||||
else:
|
else:
|
||||||
|
|
||||||
print(n)
|
print(n)
|
||||||
exit()
|
exit()
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user