from local

This commit is contained in:
dohe0342 2023-01-09 20:24:00 +09:00
parent ff62d9d54f
commit e498aa00de
2 changed files with 1 additions and 0 deletions

View File

@ -987,6 +987,7 @@ def run(rank, world_size, args):
p.requires_grad = False p.requires_grad = False
else: else:
print(n) print(n)
exit()
''' '''
for n, p in model.named_parameters(): for n, p in model.named_parameters():
if 'layer' not in n: if 'layer' not in n: