Commit 8512913
Unwrap PipelinedPostproc modules when detaching model from pipeline (meta-pytorch#2822)
Summary:
Pull Request resolved: meta-pytorch#2822
Ensure that we unwrap `PipelinedPostproc` modules when calling `detach()`. Otherwise these modules will return the cached results after unwrapping
Reviewed By: che-sh
Differential Revision: D71150731
fbshipit-source-id: f7bb172afca7f8f511fd7b96decd5e749cc5b2c51 parent e860a2c commit 8512913
File tree
3 files changed
+269
-117
lines changed- torchrec/distributed/train_pipeline
- tests
3 files changed
+269
-117
lines changed
0 commit comments