You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
In the input reinforcement branches it seem that there are redundants Average pool on the original input in the deeper stages.
Do you see any problem to pass the strided average input image after each DownSample block to the next one?
In DownSampler forward method, return input2:
defforward(self, input, input2=None):
''' :param input: input feature map :return: feature map down-sampled by a factor of 2 '''avg_out=self.avg(input)
eesp_out=self.eesp(input)
output=torch.cat([avg_out, eesp_out], 1)
ifinput2isnotNone:
#assuming the input is a square image# Shortcut connection with the input imagew1=avg_out.size(2)
whileTrue:
input2=F.avg_pool2d(input2, kernel_size=3, padding=1, stride=2)
w2=input2.size(2)
ifw2==w1:
breakoutput=output+self.inp_reinf(input2)
returnself.act(output), input2
In the input reinforcement branches it seem that there are redundants Average pool on the original input in the deeper stages.
Do you see any problem to pass the strided average input image after each DownSample block to the next one?
In DownSampler forward method, return input2:
In Espnet class, overwrite the input object:
The text was updated successfully, but these errors were encountered: