My model is Darknet / Yolo Tiny model, I didn’t change the model definition, but you can find the model definition here: https://paste.ofcode.org/YPZftRpwZFsNydkuCfNZvc
I check the code, and I think i keep the batch dimension constant all the time