Change input shape of pretrained model to process input like this [8, 30, 3, 256, 256]

Double post from here with answer.