class_docstr: Test case for the patch module.
class_docstr_tok: ["Test","case","for","the","patch","module","."]
code: def setUp(self):
      self.lr = 0.1
      self.linear = torch.nn.Linear(4, 3)
      self._model = nn.Sequential(
          nn.Linear(4, 3),
          nn.ReLU(),
          nn.Linear(3, 4, bias=None),
          nn.Sigmoid(),
          nn.Linear(4, 2),
      )
      self._shared_param_model = _Enc()
      proj = nn.Linear(4, 3)
      self._shared_param_seq_model = nn.Sequential(
          proj,
          nn.ReLU(),
          nn.Linear(3, 4),
          nn.Sigmoid(),
          proj,
          nn.Tanh(),
          nn.Linear(3, 2),
      )
      self._partially_used_model = _PartiallyUsed()
      self._batchnorm_mlp = torch.nn.Sequential(
          torch.nn.Linear(4, 3),
          torch.nn.ReLU(),
          torch.nn.BatchNorm1d(3),
          torch.nn.Linear(3, 4),
          torch.nn.Sigmoid(),
          torch.nn.BatchNorm1d(4),
          torch.nn.Linear(4, 2),
      )
  
imports: import unittest
  from parameterized import parameterized
  import torch
  from torch import nn
  import higher
label_desc: Use image node_id 1 for calling the TestPatch obj's underlying member method code with example usage: obj.setUp() without return types