class_docstr: Architecture of VisionLAN
class_docstr_tok: ["Architecture","of","VisionLAN"]
code: def __init__(
      self,
      in_channels,
      out_channels=36,
      n_layers=3,
      n_position=256,
      n_dim=512,
      max_text_length=25,
      training_step="LA",
  ):
      super(VLHead, self).__init__()
      self.MLM_VRM = MLM_VRM(
          n_layers=n_layers,
          n_position=n_position,
          n_dim=n_dim,
          max_text_length=max_text_length,
          nclass=out_channels + 1,
      )
      self.training_step = training_step
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import paddle
  from paddle import ParamAttr
  import paddle.nn
  import paddle.nn.functional
  from paddle.nn.initializer import Normal, XavierNormal
  import numpy
label_desc: Use image node_id 1 to create a new VLHead object from inherited base classes: nn with example: obj = VLHead(in_channels, out_channels, n_layers, n_position, n_dim, max_text_length, training_step)