from local

This commit is contained in:
dohe0342 2023-01-09 15:07:50 +09:00
parent 34c74b6128
commit b522b5346a
2 changed files with 1 additions and 1 deletions

View File

@ -65,7 +65,7 @@ class FairSeqData2VecEncoder(EncoderInterface):
)
model = models[0]
model.feature_grad_mult = 0.0 ## for conv network freeze
model.mask_prob = 0.5 ## for conv network freeze
model.mask_prob = 0.0 ## for conv network freeze
self.encoders = model
self.pretrained_params = copy.deepcopy(model.state_dict())