from local

This commit is contained in:
dohe0342 2023-01-10 01:11:04 +09:00
parent 2cd8fe46f5
commit bcdd15159f
2 changed files with 4 additions and 1 deletions

View File

@ -992,7 +992,10 @@ def run(rank, world_size, args):
old_name = '.'.join(layer_name_splited)
try: p.data = pre_trained_model2[old_name]
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