s444465/projekt-glebokie
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
s444501 2023-02-12 14:54:19 +01:00
parent 99b93a2579
commit e6f8958b6a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
run_translation_freezing.py
View File

@ -261,7 +261,7 @@ def freeze_model_weights(model: torch.nn.Module) -> None:
count = 0 count = 0
for param in model.parameters(): for param in model.parameters():
count += 1 count += 1
if count < 20: if count <= 20:
logger.info(f'Freezing layer {count}') logger.info(f'Freezing layer {count}')
param.requires_grad = False param.requires_grad = False
else: else: