diff --git a/orttraining/orttraining/python/training/artifacts.py b/orttraining/orttraining/python/training/artifacts.py index bc5b79eb8d183..4e76174d8255e 100644 --- a/orttraining/orttraining/python/training/artifacts.py +++ b/orttraining/orttraining/python/training/artifacts.py @@ -120,7 +120,6 @@ def __init__(self, _loss, _loss_input_names=None): self._loss_input_names = _loss_input_names def build(self, *inputs_to_loss): - # If loss_input_names is passed, only pass the specified input names to the loss function. if self._loss_input_names: inputs_to_loss = self._loss_input_names