class_docstr: PVT2FFN
class_docstr_tok: ["PVT2FFN"]
code: def __init__(self, in_features, hidden_features):
      super().__init__()
      self.fc1 = nn.Linear(in_features, hidden_features)
      self.dw_conv = DWConv(hidden_features)
      self.act = nn.GELU()
      self.fc2 = nn.Linear(hidden_features, in_features)
      self.apply(init_vit_weights)
  
imports: import torch
  from torch import nn
  from timm.models.layers import DropPath
  import numpy
  from towhee.models.wave_vit.wave_vit_utils import DWT2D, IDWT2D
  from towhee.models.utils.init_vit_weights import init_vit_weights
  from towhee.models.layers.mlp import Mlp
label_desc: Use image node_id 1 to create a new PVT2FFN object from inherited base classes: nn with example: obj = PVT2FFN(in_features, hidden_features)