code: def main():
      """
      Using vanilla Lightning API to train/test
      """
      data = MNISTDataModule(batch_size=64)
      model = LitSampleConvNetClassifier()
  
      dp_data = DPLightningDataModule(data)
  
      trainer = pl.Trainer(
          max_epochs=10,
          enable_model_summary=False,
      )
      trainer.fit(model, dp_data)
  
      trainer.test(model, data)
      trainer.test(model, dp_data)
  
imports: import os
  import warnings
  import pytorch_lightning
  import torch.nn
  import torch.nn.functional
  import torch.optim
  import torchmetrics
  from opacus import PrivacyEngine
  from opacus.data_loader import DPDataLoader
  from opacus.lightning import DPLightningDataModule
  from pl_bolts.datamodules import MNISTDataModule
  from pytorch_lightning.utilities.cli import LightningCLI
label_desc: Use image node_id 1 for calling a global function with example usage: main() without return types