methylbert icon indicating copy to clipboard operation
methylbert copied to clipboard
verified publisher icon CompEpigen

Parameter name inconsistency: masked_lm_labels vs labels in forward method

Open sw1125598670 opened this issue 2 months ago • 0 comments

Hello! I encountered an issue during pretraining where the forward method doesn't recognize the masked_lm_labels parameter.

"/home/mickey/Tools/miniconda3/envs/methylbert/lib/python3.11/site-packages/methylbert/trainer.py", line 287 mask_lm_output = self.model.forward(input_ids = data["input"], masked_lm_labels = data["label"]) Questions: Should the parameter name be labels instead of masked_lm_labels?

sw1125598670 avatar Nov 26 '25 11:11 sw1125598670