code: def get_lenet5madry_with_width(widen_factor):
      return LeNet5Madry(
          nb_filters=(
              1,
              int(widen_factor * 32),
              int(widen_factor * 64),
          ),
          nb_hiddens=(
              7 * 7 * int(widen_factor * 64),
              int(widen_factor * 1024),
          ),
      )
  
imports: import math
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 for calling a global function with example usage: get_lenet5madry_with_width(widen_factor) and returns: LeNet5Madry