search

PeterL1n / RobustVideoMatting

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

隐状态在训练过程中的调用位置？ #153

Closed nicolaus-huang closed 2 years ago

nicolaus-huang commented 2 years ago

你好,我想问一下在训练过程中是否没有使用隐状态? P$S763H7{ALZ2OM7HMGS0 5

而在推理中, AP9GAKRGH%PZSC@C2F8V@OL

rec存储了隐状态

PeterL1n commented 2 years ago

用了。训练时的batch是五维BTCHW。模型内部循环使用隐藏值。