diff --git a/python/tvm/relay/frontend/darknet.py
b/python/tvm/relay/frontend/darknet.py
index 936d7c0dc..62a320780 100644
--- a/python/tvm/relay/frontend/darknet.py
+++ b/python/tvm/relay/frontend/darknet.py
@@ -637,12 +637,12 @@ class GraphProto(object):
attr.update({'coords' : layer.coords})
attr.update({'background' : layer.background})
attr.update({'softmax' : layer.softmax})
- attr.update({'shape' : (1, layer.c, layer.h, layer.w)})
+ attr.update({'shape' : (-1, layer.c, layer.h, layer.w)})
elif LAYERTYPE.YOLO == layer_type:
attr.update({'n' : layer.n})
attr.update({'classes' : layer.classes})
- attr.update({'shape' : (1, layer.c, layer.h, layer.w)})
+ attr.update({'shape' : (-1, layer.c, layer.h, layer.w)})
elif LAYERTYPE.UPSAMPLE == layer_type:
attr.update({'scale' : layer.stride})
You can apply this diff to `python/tvm/relay/frontend/darknet.py` to quickly
solve this issue.
This will change your output to
module.get_output(0) --> (4, 255, 26, 26)
module.get_output(1) --> (4, 255, 13, 13)
But you need to do the post processing to consider the batch values as well.
`layer_out['output'] = layer_out['output'].reshape(out_shape)`
Here the outshape must consider the batch size as well, currently its written
to handle only batch_size of 1.
---
[Visit
Topic](https://discuss.tvm.ai/t/yolov3-tiny-batch-input-test-failed/6796/2) to
respond.
You are receiving this because you enabled mailing list mode.
To unsubscribe from these emails, [click
here](https://discuss.tvm.ai/email/unsubscribe/8b2cf41b4fd4ce669a684f4f805ad3eb6acae2b196a52527802c50183a771ac3).
