code: def WideResnetGroup(
      n, channels, strides=(1, 1), bn_momentum=0.9, mode="train"
  ):
      shortcut = [
          tl.Conv(channels, (3, 3), strides, padding="SAME"),
      ]
      return [
          tl.Residual(
              WideResnetBlock(
                  channels, strides, bn_momentum=bn_momentum, mode=mode
              ),
              shortcut=shortcut,
          ),
          tl.Residual(
              [
                  WideResnetBlock(
                      channels,
                      (1, 1),
                      bn_momentum=bn_momentum,
                      mode=mode,
                  )
                  for _ in range(n - 1)
              ]
          ),
      ]
  
imports: from trax import layers
label_desc: Use image node_id 5 for calling a global function with example usage: WideResnetGroup(n, channels, strides, bn_momentum, mode) and returns: list