from local

This commit is contained in:
dohe0342 2023-01-08 18:06:33 +09:00
parent 8f20ba4bd7
commit fc64f1f842
2 changed files with 2 additions and 1 deletions

View File

@ -980,7 +980,8 @@ def run(rank, world_size, args):
old_name = '.'.join(layer_name_splited) old_name = '.'.join(layer_name_splited)
try: p.data = pre_trained_model[old_name] try: p.data = pre_trained_model[old_name]
except: print(f'pre-trained model has no parameter named {n}.') except: print(f'pre-trained model has no parameter named {n}.')
print('.'.join(layer_name_splited)) else:
print(f'skipping param load {n}')
exit() exit()