davda54 commited on
Commit
36bd0b8
1 Parent(s): 77fb7a3

Upload modeling_ltgbert.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_ltgbert.py +4 -2
modeling_ltgbert.py CHANGED
@@ -252,8 +252,10 @@ class LtgbertPreTrainedModel(PreTrainedModel):
252
  elif isinstance(module, nn.Embedding):
253
  nn.init.trunc_normal_(module.weight.data, mean=0.0, std=std, a=-2*std, b=2*std)
254
  elif isinstance(module, nn.LayerNorm):
255
- module.bias.data.zero_()
256
- module.weight.data.fill_(1.0)
 
 
257
 
258
 
259
  class LtgbertModel(LtgbertPreTrainedModel):
 
252
  elif isinstance(module, nn.Embedding):
253
  nn.init.trunc_normal_(module.weight.data, mean=0.0, std=std, a=-2*std, b=2*std)
254
  elif isinstance(module, nn.LayerNorm):
255
+ if module.bias is not None:
256
+ module.bias.data.zero_()
257
+ if module.weight is not None:
258
+ module.weight.data.fill_(1.0)
259
 
260
 
261
  class LtgbertModel(LtgbertPreTrainedModel):