Non-local_pytorch

Non-local_pytorch copied to clipboard

net.train() again in demo_MNIST_train.py file

2

Hi, I started to learn pytorch a few months ago, so I might be questioning basic stuff. in [[Non-local_pytorch](https://github.com/AlexHex7/Non-local_pytorch)/demo_MNIST_train.py](url) file, there was `net.train()` in the 50th line, within the training...

Dylee-y

https://github.com/AlexHex7/Non-local_pytorch/blame/ec15789af4533448169f3dc36abe542badd34221/nl_map_vis/nl_map_vis.py#L33 > nl_map = cv2.resize(nl_map, dsize=(h, w)) Shouldn't it be ` dsize=(w, h)` ? I encountered this bug when my image size W ≠ H.

onlyairnopods