Hongzhehao

@SoulCoderCN 目前看来应该是需要手动调整的，只是我还没有找到相关联的地方在哪里

@SoulCoderCN 请问下你的显存是多少呀，我之前运行的时候，600尺度的图片 一直是 cuda out of memory