I am trying to check the values for the input parameters from the data I have been able to find out values for  the function predict_transform.
inp_dim=608
anchors=9
num_classes=80
But here the problem lies I am not able to determine prediction.size(0)
x = x.data

so I need to size(0) of this x… for that I have tried printing it in the function so for batch_size=prediction.size(0) I get 1
stride is 11
grid size is around 55
bbox_attrs is 85
num_anchors=9
Now what might be the error??
edit: I got value of batch_size
by adding print(batch_size) in predict_transform before the line which produces error same for stride…