Skip to content
This repository was archived by the owner on Jan 6, 2023. It is now read-only.

Commit bc88e69

Browse files
Pyre Bot Jrfacebook-github-bot
Pyre Bot Jr
authored andcommitted
suppress errors in pytorch/elastic
Differential Revision: D37172465 fbshipit-source-id: 452c00cd54d4682a9512e0a0842d9526f7bc8fc8
1 parent 7f9003d commit bc88e69

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/imagenet/main.py

+2
Original file line numberDiff line numberDiff line change
@@ -372,6 +372,8 @@ def load_checkpoint(
372372
print(f"=> checkpoint broadcast size is: {blob_len}")
373373

374374
if rank != max_rank:
375+
# pyre-fixme[6]: For 1st param expected `Union[List[int], Size,
376+
# typing.Tuple[int, ...]]` but got `Union[bool, float, int]`.
375377
blob = torch.zeros(blob_len.item(), dtype=torch.uint8)
376378
else:
377379
blob = torch.as_tensor(raw_blob, dtype=torch.uint8)

0 commit comments

Comments
 (0)