class_docstr: DRAEM PyTorch model consisting of the reconstructive and discriminative sub networks.
class_docstr_tok: ["DRAEM","PyTorch","model","consisting","of","the","reconstructive","and","discriminative","sub","networks","."]
code: def __init__(self, sspcab: bool = False) -> None:
      super().__init__()
      self.reconstructive_subnetwork = ReconstructiveSubNetwork(
          sspcab=sspcab
      )
      self.discriminative_subnetwork = DiscriminativeSubNetwork(
          in_channels=6, out_channels=2
      )
  
imports: from __future__ import annotations
  import torch
  from torch import Tensor, nn
  from anomalib.models.components.layers import SSPCAB
label_desc: Use image node_id 1 to create a new DraemModel object from inherited base classes: nn with example: obj = DraemModel(sspcab)