class_docstr: WideResNet block.
code: def __call__(self, x: JaxArray, training: bool) -> JaxArray:
      o1 = objax.functional.relu(self.norm_1(x, training))
      y = self.conv_1(o1)
      o2 = objax.functional.relu(self.norm_2(y, training))
      z = self.conv_2(o2)
      return z + self.proj_conv(o1) if self.proj_conv else z + x
  
file_path: objax/objax/zoo/wide_resnet.py
imports: import functools
  from typing import Callable, List
  import objax
  from objax.typing import JaxArray
label_desc: Use image node_id 2 for calling the WRNBlock obj's underlying member method code with example usage: obj.__call__(x, training) and returns: unknown