[Bug] nn.Conv3D groups not working correctly #17062
Labels
needs-triage
PRs or issues that need to be investigated by maintainers to find the right assignees to address it
type: bug
nn.Conv3D
is buggedtvm/python/tvm/relax/frontend/nn/modules.py
Lines 314 to 325 in 78a1f80
line 325 should use
in_channels
instead ofself.in_channels
to make groups work. Otherwisein_channels
calculated at line 317 would not be effectively used in weight parameters.The text was updated successfully, but these errors were encountered: