code: def __init__(self, context):
      self.context = context
  
      self.model = context.wrap_model(nn.Linear(1, 1, False))
      self.opt = context.wrap_optimizer(
          torch.optim.SGD(self.model.parameters(), lr=0.001),
          backward_passes_per_step=2,
      )
  
file_path: determined/harness/tests/experiment/fixtures/ancient-checkpoints/0.17.6-pytorch/code/model_def.py
imports: import torch
  from torch import nn
  from determined import pytorch
label_desc: Use image node_id 1 to create a new OneVarPytorchTrial object from inherited base classes: pytorch with example: obj = OneVarPytorchTrial(context)