fix

skeleton_modeling_roberta.py

@@ -58,7 +58,7 @@ def SkeletonRobertaLayer(layer_id,layer,hidden,interventions):
     new_hidden =  layer.output.LayerNorm(hidden_post_interm_res) # layer_norm
     return new_hidden
 
-def SkeletonBertForMaskedLM(model,input_ids,interventions):
+def SkeletonRobertaForMaskedLM(model,input_ids,interventions):
     core_model = model.roberta
     lm_head = model.lm_head
     output_hidden = []