class_docstr: Wrapper around a `generator` and a `critic` to create a GAN.
class_docstr_tok: ["Wrapper","around","a","`","generator","`","and","a","`","critic","`","to","create","a","GAN","."]
code: def __init__(
      self,
      generator: nn.Module = None,  # The generator PyTorch module
      critic: nn.Module = None,  # The discriminator PyTorch module
      gen_mode: None
      | bool = False,  # Whether the GAN should be set to generator mode
  ):
      if generator is not None:
          self.generator = generator
      if critic is not None:
          self.critic = critic
      store_attr("gen_mode")
  
imports: from __future__ import annotations
  from ..basics import *
  from .all import *
label_desc: Use image node_id 1 to create a new GANModule object from inherited base classes: Module with example: obj = GANModule(generator, critic, gen_mode)