Commit 047760b
add train pipeline class in logging (meta-pytorch#3486)
Summary:
Pull Request resolved: meta-pytorch#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: 710a84cdb611cbaa2d255e40fb8464ee3826dac31 parent 3a6cf2e commit 047760b
1 file changed
+2
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
110 | 110 | | |
111 | 111 | | |
112 | 112 | | |
| 113 | + | |
| 114 | + | |
113 | 115 | | |
114 | 116 | | |
115 | 117 | | |
| |||
0 commit comments