Skip to content

Commit 749de9a

Browse files
Update listing-2-6.txt
1 parent aee37cf commit 749de9a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

code/chapter-2/listing-2-6.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
if get_worker_rank() == 0: #A
2-
create_and_send_shards(dataset) #A
2+
create_and_send_shards(dataset) #A
33
shard = read_next_shard_locally() #B
44
while shard is not None:
55
model.train(shard) #C
6-
shard = read_next_shard_locally() #D
6+
shard = read_next_shard_locally() #D
77

88
#A Create and send shards to all other worker machines from the worker machine with rank 0.
99
#B Read the next shard available locally in this worker machine.

0 commit comments

Comments
 (0)