mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 12:50:06 +06:00
Merge 1a8d584598
into 2d561713f8
This commit is contained in:
commit
cdb2f0aba0
@ -341,7 +341,7 @@ class SwinPatchMerging(nn.Module):
|
||||
batch_size, dim, num_channels = input_feature.shape
|
||||
|
||||
input_feature = input_feature.view(batch_size, height, width, num_channels)
|
||||
# pad input to be disible by width and height, if needed
|
||||
# pad input to be divisible by width and height, if needed
|
||||
input_feature = self.maybe_pad(input_feature, height, width)
|
||||
# [batch_size, height/2, width/2, num_channels]
|
||||
input_feature_0 = input_feature[:, 0::2, 0::2, :]
|
||||
|
@ -185,7 +185,7 @@ class PipelineChunkIterator(PipelineIterator):
|
||||
# Try to return next item
|
||||
processed = next(self.subiterator)
|
||||
except StopIteration:
|
||||
# When a preprocess iterator ends, we can start lookig at the next item
|
||||
# When a preprocess iterator ends, we can start looking at the next item
|
||||
# ChunkIterator will keep feeding until ALL elements of iterator
|
||||
# all have created their subiterator and have been iterating against.
|
||||
#
|
||||
|
Loading…
Reference in New Issue
Block a user