You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Set the model to the inference mode by default + ensure training script is not impacted + update failing unit-tests
Remove the hf-format from Next Item Prediction task + ensure training / inference are not impacted + update failing unit-tests
Replace ignore_masking with testing flag + ensure training / inference are not impacted + update failing unit-tests
P.S: This PR removes the hf_format only for NextItemPredictionTask. Other tasks such as BinaryClassificationTask and RegressionTask still use hf_format as removing the flag for them at this stage does not make sense for them, they use the PredictionTask.forward() only for inference and use compute_loss() for training and eval.
The text was updated successfully, but these errors were encountered:
This part includes 3 tasks:
Set the model to the inference mode by default + ensure training script is not impacted + update failing unit-tests
Remove the hf-format from Next Item Prediction task + ensure training / inference are not impacted + update failing unit-tests
Replace ignore_masking with testing flag + ensure training / inference are not impacted + update failing unit-tests
P.S: This PR removes the
hf_format
only forNextItemPredictionTask
. Other tasks such asBinaryClassificationTask
andRegressionTask
still usehf_format
as removing the flag for them at this stage does not make sense for them, they use thePredictionTask.forward()
only for inference and usecompute_loss()
for training and eval.The text was updated successfully, but these errors were encountered: