Pytorch-HarDNet
Pytorch-HarDNet copied to clipboard
I think this function doesn't make it flatten.
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))