code: def __init__(
      self,
      prediction_length: int,
      context_length: int,
  ) -> None:
      super().__init__()
  
      self.prediction_length = prediction_length
      self.context_length = context_length
      # self.criterion = nn.MSELoss(reduction='none')
      self.criterion = nn.SmoothL1Loss(reduction="none")
  
      modules = []
      modules.append(nn.Linear(context_length, prediction_length))
      self.linear = nn.Sequential(*modules)
  
file_path: gluonts/src/gluonts/nursery/SCott/model/ar/ar_network.py
imports: from typing import List
  import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new ARNetworkBase object from inherited base classes: nn with example: obj = ARNetworkBase(prediction_length, context_length)