From ffe1af777d409f158f52cdaf06436e77a97cfdd4 Mon Sep 17 00:00:00 2001 From: Qingyi Lu Date: Thu, 31 Oct 2024 16:13:35 -0700 Subject: [PATCH] Bulk Eval MAST Run Pipeline on APS Summary: Bulk Eval MAST Run Pipeline on APS Differential Revision: D65301821 --- torchrec/distributed/train_pipeline/train_pipelines.py | 1 - 1 file changed, 1 deletion(-) diff --git a/torchrec/distributed/train_pipeline/train_pipelines.py b/torchrec/distributed/train_pipeline/train_pipelines.py index 2886b23eb..50cbc1ffe 100644 --- a/torchrec/distributed/train_pipeline/train_pipelines.py +++ b/torchrec/distributed/train_pipeline/train_pipelines.py @@ -376,7 +376,6 @@ def __init__( self._dataloader_iter: Optional[Iterator[In]] = None self._dataloader_exhausted: bool = False self._context_type: Type[TrainPipelineContext] = context_type - self._model_fwd: Callable[[Optional[In]], Tuple[torch.Tensor, Out]] = ( custom_model_fwd if custom_model_fwd else model )