from local
This commit is contained in:
parent
2cd8fe46f5
commit
bcdd15159f
Binary file not shown.
@ -992,7 +992,10 @@ def run(rank, world_size, args):
|
|||||||
old_name = '.'.join(layer_name_splited)
|
old_name = '.'.join(layer_name_splited)
|
||||||
try: p.data = pre_trained_model2[old_name]
|
try: p.data = pre_trained_model2[old_name]
|
||||||
except: print(f'pre-trained model has no parameter named {n}.')
|
except: print(f'pre-trained model has no parameter named {n}.')
|
||||||
|
del pre_trained1
|
||||||
|
del pre_trained2
|
||||||
|
del pre_trained_model1
|
||||||
|
del pre_trained_model2
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user