code: def Axial_50s(pretrained=False, **kwargs):
      model = AxialAttentionNet(
          AxialBlock, [3, 4, 6, 3], s=0.5, **kwargs
      )
      return model
  
imports: import math
  import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 3 for calling a global function with example usage: Axial_50s(pretrained) and returns: model