Split train and compute functions in VideoParser