code: def __init__(self, weight):
      super(Mul, self).__init__()
      self.weight = weight
  
imports: from argparse import ArgumentParser
  from typing import List
  import time
  import numpy
  from tqdm import tqdm
  import torch
  from torch.cuda.amp import GradScaler, autocast
  from torch.nn import CrossEntropyLoss, Conv2d, BatchNorm2d
  from torch.optim import SGD, lr_scheduler
  import torchvision
  from fastargs import get_current_config, Param, Section
  from fastargs.decorators import param
  from fastargs.validation import And, OneOf
  from ffcv.fields import IntField, RGBImageField
  from ffcv.fields.decoders import IntDecoder, SimpleRGBImageDecoder
  from ffcv.loader import Loader, OrderOption
  from ffcv.pipeline.operation import Operation
  from ffcv.transforms import RandomHorizontalFlip, Cutout, RandomTranslate, Convert, ToDevice, ToTensor, ToTorchImage
  from ffcv.transforms.common import Squeeze
  from ffcv.writer import DatasetWriter
label_desc: Use image node_id 1 to create a new Mul object from inherited base classes: ch.nn with example: obj = Mul(weight)