Arnold
Arnold copied to clipboard
AssertionError&RuntimeError bug fixes for PyTorch1.7.0 implementation
The current version of the source code does not work properly on some PyTorch versions.
I have tried on the current latest PyTorch version (v1.7.0) at first and faced with the AssertionError which was mentioned in issue #8. Thanks to PR #10, AssertionError was fixed. However, the code was not able to work properly on PyTorch1.7.0, due to a RuntimeError.
This PR provides a valid code, tested on the current latest version of PyTorch.
RuntimeError: torch.cat(): Tensors must have same number of dimensions: got 2 and 3 Now I got this error, could you give me some advice?