class_docstr: Test CNNModule.
class_docstr_tok: ["Test","CNNModule","."]
code: def setup_method(self):
      self.batch_size = 64
      self.input_width = 32
      self.input_height = 32
      self.in_channel = 3
      self.dtype = torch.float32
      self.input_spec = InOutSpec(
          akro.Box(
              shape=[
                  self.in_channel,
                  self.input_height,
                  self.input_width,
              ],
              high=np.inf,
              low=np.inf,
          ),
          None,
      )
      self.input = torch.zeros(
          (
              self.batch_size,
              self.in_channel,
              self.input_height,
              self.input_width,
          ),
          dtype=self.dtype,
      )
  
file_path: garage/tests/garage/torch/modules/test_cnn_module.py
imports: import pickle
  import akro
  import numpy
  import pytest
  import torch
  import torch.nn
  from garage import InOutSpec
  from garage.torch.modules import CNNModule
  from garage.torch.modules.cnn_module import _check_spec
label_desc: Use image node_id 1 for calling the TestCNNModule obj's underlying member method code with example usage: obj.setup_method() without return types