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
Summary:
Pull Request resolved: #3486
# context
* add TrainPipeline class logging to help debugging which train pipeline is used in the training job
* this will produce the following log
* the logging is done once when the train pipeline class is initiated (per rank)
```
INFO:torchrec.distributed.train_pipeline.train_pipelines:TrainPipeline class: <class 'torchrec.distributed.train_pipeline.train_pipelines.TrainPipelineSparseDist'>
```
Reviewed By: nipung90
Differential Revision: D85577739
fbshipit-source-id: 710a84cdb611cbaa2d255e40fb8464ee3826dac3
0 commit comments