code: def Classifier(in_features, out_features, is_nonlinear=False):
      if is_nonlinear:
          return torch.nn.Sequential(
              torch.nn.Linear(in_features, in_features // 2),
              torch.nn.ReLU(),
              torch.nn.Linear(in_features // 2, in_features // 4),
              torch.nn.ReLU(),
              torch.nn.Linear(in_features // 4, out_features),
          )
      else:
          return torch.nn.Linear(in_features, out_features)
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  import torchvision.models
label_desc: Use image node_id 1 for calling a global function with example usage: Classifier(in_features, out_features, is_nonlinear) and returns: torch, torch