Ok,
I have found an explanation here if someone else is interested. I have overseen the actual error.
Cannot insert a Tensor that requires grad as a constant - jit - PyTorch Forums.
Ok,
I have found an explanation here if someone else is interested. I have overseen the actual error.
Cannot insert a Tensor that requires grad as a constant - jit - PyTorch Forums.