from local

This commit is contained in:
dohe0342 2022-12-10 14:12:41 +09:00
parent bc4a865377
commit db5f76ce67
2 changed files with 1 additions and 1 deletions

View File

@ -556,7 +556,7 @@ def get_encoder_model(params: AttributeDict) -> nn.Module:
w2v_url='None', w2v_url='None',
output_size=params.encoder_dim, output_size=params.encoder_dim,
freeze_finetune_updates=params.freeze_finetune_updates*params.accum_grads, freeze_finetune_updates=params.freeze_finetune_updates*params.accum_grads,
additional_block=True, #params.additional_block, additional_block=params.additional_block,
) )
return encoder return encoder