Skip to content

Commit 465b52b

Browse files
committed
fix example code
1 parent 121d3b3 commit 465b52b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

doc/api/training/sdp_versions/latest/smd_data_parallel_pytorch.rst

+1-1
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ you will have for distributed training with the distributed data parallel librar
112112
def main():
113113
114114
    # Scale batch size by world size
115-
    batch_size //= dist.get_world_size() // 8
115+
    batch_size //= dist.get_world_size()
116116
    batch_size = max(batch_size, 1)
117117
118118
    # Prepare dataset

0 commit comments

Comments
 (0)