jhjacobsen/pytorch-i-revnet

Potential bug in model_utils.py

danfeiX opened this issue · 1 comments

Thanks for the great work. I think I found a potential bug in your code

https://github.com/jhjacobsen/pytorch-i-revnet/blob/master/models/model_utils.py#L36

Shouldn't it be return x[:, :x.size(1) - self.pad_size, :, :]?

Thanks for reporting the issue, you are right and I'll fix it!
Just for the record: It does not affect the reported results.