提交 bb831c3b 编写于 作者: G gaotingquan 提交者: cuicheng01

code style

上级 70a784ce
...@@ -479,7 +479,7 @@ class LinearAttnFFN(nn.Layer): ...@@ -479,7 +479,7 @@ class LinearAttnFFN(nn.Layer):
return x return x
class MobileViTV3Block_v2(nn.Layer): class MobileViTV3BlockV2(nn.Layer):
""" """
This class defines the `MobileViTV3 block` This class defines the `MobileViTV3 block`
""" """
...@@ -749,7 +749,7 @@ class MobileViTV3(nn.Layer): ...@@ -749,7 +749,7 @@ class MobileViTV3(nn.Layer):
if self.mobilevit_v2_based: if self.mobilevit_v2_based:
block.append( block.append(
MobileViTV3Block_v2( MobileViTV3BlockV2(
in_channels=input_channel, in_channels=input_channel,
attn_unit_dim=cfg["attn_unit_dim"], attn_unit_dim=cfg["attn_unit_dim"],
ffn_multiplier=cfg.get("ffn_multiplier"), ffn_multiplier=cfg.get("ffn_multiplier"),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册