Significant RAM usage increase on performing a get batch of 4K video

[svats2k]

decord version (installed via pip): 0.6.0 OS: Ubuntu 20.04

Big bunny video downloaded from https://4kmedia.org/big-buck-bunny-4k-demo/

My code is the following:

import decord
vr = decord.VideoReader(<local file path>)
batch = vr.get_batch(range(3))

as soon as I try to get a batch of frames, the RAM usage builds up beyond 30 GB and the process gets killed.

[svats2k]

I do not observe this issue when I use decord frame seek as a part of a python application. However, when I seek a frame in a jupyter notebook session or in a repl session I observe the memory issue.

[sueskind]

As a workaround people in #208 suggested doing vr.seek(0) after the vr.get_batch(), which works for me. However, a fix for this would be much appreciated still.

[tuanlda78202]

Thank you so much, it works for me!