File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed
_modules/torch/utils/data Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -1159,7 +1159,10 @@ <h1>Source code for torch.utils.data.dataloader</h1><div class="highlight"><pre>
11591159
11601160 < span class ="c1 "> # Exit workers now.</ span >
11611161 < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> workers_done_event</ span > < span class ="o "> .</ span > < span class ="n "> set</ span > < span class ="p "> ()</ span >
1162- < span class ="k "> for</ span > < span class ="n "> worker_id</ span > < span class ="ow "> in</ span > < span class ="nb "> range</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> num_workers</ span > < span class ="p "> ):</ span >
1162+ < span class ="k "> for</ span > < span class ="n "> worker_id</ span > < span class ="ow "> in</ span > < span class ="nb "> range</ span > < span class ="p "> (</ span > < span class ="nb "> len</ span > < span class ="p "> (</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> workers</ span > < span class ="p "> )):</ span >
1163+ < span class ="c1 "> # Get number of workers from `len(self.workers)` instead of</ span >
1164+ < span class ="c1 "> # `self.num_workers` in case we error before starting all</ span >
1165+ < span class ="c1 "> # workers.</ span >
11631166 < span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> workers_status</ span > < span class ="p "> [</ span > < span class ="n "> worker_id</ span > < span class ="p "> ]:</ span >
11641167 < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> _shutdown_worker</ span > < span class ="p "> (</ span > < span class ="n "> worker_id</ span > < span class ="p "> )</ span >
11651168 < span class ="k "> for</ span > < span class ="n "> w</ span > < span class ="ow "> in</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> workers</ span > < span class ="p "> :</ span >
You can’t perform that action at this time.
0 commit comments