Python Numpy sum error (single-shot multibox object detector in PyTorch)

https://stackoverflow.com/questions/46420007/numpy-sum-got-an-keepdims-error

By updating PyTorch version to 0.2 > problem is solved

https://discuss.pytorch.org/t/updating-pytorch/309