参考pointnet++语义分割部分的 train.py 文件中的 train() 部分代码。
思路是:
一方面,可以固定轮数更新模型,防止模型中断;
另一方面,通过比较 测试集准确度 的最高值,保存模型,即为最佳模型。