code: def forward(self, x):
      x = self.relu1(self.conv1(x))
      x = self.max_pool1(x)
      x = self.relu2(self.conv2(x))
      x = self.max_pool2(x)
      x = x.view(-1, x.size()[1:].numel())
      x = self.acti(self.fc1(x), **self.acti_kw)
      # x = F.relu6(self.fc1(x))
      x = self.fc2(x)
      return F.log_softmax(x, dim=1)
  
file_path: nni/test/algo/compression/pruning/test_speedup_softmax.py
imports: import unittest
  import torch
  import torch.nn.functional
  from nni.compression.utils.counter import count_flops_params
  from nni.compression.speedup import ModelSpeedup
  from nni.compression.pruning import L1NormPruner
label_desc: Use image node_id 2 for calling the NaiveModel obj's underlying member method code with example usage: obj.forward(x) and returns: F