PeterL1n / RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
https://peterl1n.github.io/RobustVideoMatting/
GNU General Public License v3.0
8.32k stars 1.11k forks source link

为什么epoch-end最大只到28? #247

Closed surifans closed 10 months ago

surifans commented 11 months ago

发现训练过程中epoch-end最大是28,为什么不设置大一些? 后面几个阶段的训练只有几个epoch,这样训练会充分嘛? 看起来训练数据量挺大的,epoch-end设的太小了,会不会造成模型不够收敛呢? 好吧,其实我有点怀疑,如果epoch-end设太大,每个阶段训太多,是不是会造成时序模块退化呢?毕竟如果不用时序模块,仅用单帧分割,模型似乎也能得到比较好的效果吧?既然模型单帧分割能够得到差不太多的效果,那为啥要去用时序特征?可能训太久了,在某几个batch开始,模型就发现仅用当前帧提取的特征就能够实现分割,时序提供的特征是多余的,所以模型可能会退化吧?