Skip to content

Commit af68d68

Browse files
mrshenliholly1238
andauthored
Fixing typo in distributed pipeline tutorial (pytorch#1332)
Co-authored-by: holly1238 <[email protected]>
1 parent ff0bfef commit af68d68

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/dist_pipeline_parallel_tutorial.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ simplify distributed optimizer construction, which will be used later.
225225
226226
def forward(self, xs):
227227
out_futures = []
228-
for x in iter(xs.split(self.split_size, dim=0)):
228+
for x in iter(xs.split(self.num_split, dim=0)):
229229
x_rref = RRef(x)
230230
y_rref = self.p1_rref.remote().forward(x_rref)
231231
z_fut = self.p2_rref.rpc_async().forward(y_rref)

0 commit comments

Comments
 (0)