RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

[3XiangyiKong3]

This seems to be a problem when running uq_models, but it worked after changing from .view() to .reshape()