显存溢出怎么办？

[XinyiYing]

训练显存溢出可以通过调小batch size和patch size降低显存占用。
测试显存溢出可以通过调小batch [size，裁块推理拼接降低显存占用，代码参考LESPS的test.py。