Skip to content

Commit a7905a5

Browse files
committed
Fix style issue
1 parent 4392454 commit a7905a5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

segmentation_models_pytorch/encoders/mix_transformer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
#
44
# Licensed under the NVIDIA Source Code License. For full license
55
# terms, please refer to the LICENSE file provided with this code
6-
# or visit NVIDIA's official repository at
6+
# or visit NVIDIA's official repository at
77
# https://github.com/NVlabs/SegFormer/tree/master.
88
#
99
# This code has been modified.
@@ -31,6 +31,7 @@ def forward(self, x):
3131
)
3232
return x
3333

34+
3435
class Mlp(nn.Module):
3536
def __init__(
3637
self,
@@ -466,7 +467,6 @@ def reset_classifier(self, num_classes, global_pool=""):
466467
)
467468

468469
def forward_features(self, x):
469-
B = x.shape[0]
470470
outs = []
471471

472472
# stage 1

0 commit comments

Comments
 (0)