Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit 86bd6c7

Browse files
resolving delete conflicts
2 parents d2ac487 + ca2baff commit 86bd6c7

2 files changed

Lines changed: 0 additions & 2 deletions

File tree

examples/imagenet/main_amp.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,6 @@ def fast_collate(batch):
8282
tensor = torch.zeros( (len(imgs), 3, h, w), dtype=torch.uint8 )
8383
for i, img in enumerate(imgs):
8484
nump_array = np.asarray(img, dtype=np.uint8)
85-
tens = torch.from_numpy(nump_array)
8685
if(nump_array.ndim < 3):
8786
nump_array = np.expand_dims(nump_array, axis=-1)
8887
nump_array = np.rollaxis(nump_array, 2)

tests/L1/common/main_amp.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,6 @@ def fast_collate(batch):
8686
tensor = torch.zeros( (len(imgs), 3, h, w), dtype=torch.uint8 )
8787
for i, img in enumerate(imgs):
8888
nump_array = np.asarray(img, dtype=np.uint8)
89-
tens = torch.from_numpy(nump_array)
9089
if(nump_array.ndim < 3):
9190
nump_array = np.expand_dims(nump_array, axis=-1)
9291
nump_array = np.rollaxis(nump_array, 2)

0 commit comments

Comments
 (0)