docstr: Testing PL Module.
  
  Use as follows:
  - subclass
  - modify the behavior for what you want
  
  class TestModel(BaseTestModel):
      def training_step(...):
          # do your own thing
  
  or:
  
  model = BaseTestModel()
  model.training_epoch_end = None
docstr_tok: ["Testing","PL","Module",".","Use","as","follows",":","-","subclass","-","modify","the","behavior","for","what","you","want","class","TestModel","(","BaseTestModel",")",":","def","training_step","(","...",")",":","#","do","your","own","thing","or",":","model","=","BaseTestModel","(",")","model.training_epoch_end","=","None"]
code: def __init__(self):
      """Testing PL Module.
  
      Use as follows:
      - subclass
      - modify the behavior for what you want
  
      class TestModel(BaseTestModel):
          def training_step(...):
              # do your own thing
  
      or:
  
      model = BaseTestModel()
      model.training_epoch_end = None
  
      """
      super().__init__()
      self.layer = torch.nn.Linear(32, 2)
  
imports: from typing import Optional
  import torch
  from pytorch_lightning import LightningDataModule, LightningModule
  from torch.utils.data import DataLoader, Dataset, Subset
label_desc: Use image node_id 1 to create a new BoringModel object from inherited base classes: LightningModule with example: obj = BoringModel()