qrcat

Your PyTorch version(12.1) mismatches your CUDA version(11.5), please downgrade your pytorch.

The flow is compute by `flow_2d` as input, and it just use volume render to display `flow_2d`.

It seem to KNN's problem. If you disable rigid loss, it will be faster.