Tensor size error - Githubissues

tjiagoM / stochastic-YOLO

Stochastic-YOLO: Efficient Probabilistic Object Detection under Dataset Shifts

Apache License 2.0

16 stars 6 forks source link

Tensor size error #1

Open sharifsagar80 opened 3 years ago

sharifsagar80 commented 3 years ago

Hi, I am trying to use your method to use in yolov4. but every time I am getting runtime error in training such as sizes of tensors must match except in dimension 2. Can you help me with understanding this problem?

tjiagoM commented 3 years ago

Hi, can you be more specific? :)

What error are you getting and where? What changes did you introduce in the code? Do you have any working solution for me to replicate?