Skip to content

Commit

Permalink
fix segmenter ncnn inference failure brought by open-mmlab#477
Browse files Browse the repository at this point in the history
  • Loading branch information
RunningLeon committed Jun 17, 2022
1 parent 4240f5a commit 00af7f0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mmdeploy/pytorch/functions/linear.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ def linear__ncnn(

dim = input.dim()

if dim == 2 or dim == 3 and input.shape[0] == 1:
if dim == 2:
return origin_func(input, weight, bias)
else:
out = origin_func(input, weight)
Expand Down

0 comments on commit 00af7f0

Please sign in to comment.