Pytorch-HarDNet icon indicating copy to clipboard operation
Pytorch-HarDNet copied to clipboard

I think this function doesn't make it flatten.

Open gaussian37 opened this issue 5 years ago • 0 comments

https://github.com/PingoLH/Pytorch-HarDNet/blob/12b140c4f9566d2c08a604a093879ddaf7516541/hardnet.py#L10

Is it right to revise this function to (x.view(x.data.size(0), -1) → (x.view(-1))

gaussian37 avatar Jun 08 '20 09:06 gaussian37