docstr: Training step for Pyro training.
docstr_tok: ["Training","step","for","Pyro","training","."]
code: def training_step(self, batch, batch_idx):
      """Training step for Pyro training."""
      loss = self.loss_fn(*batch)
      # Logging to TensorBoard by default
      self.log("train_loss", loss)
      return loss
  
file_path: pyro/examples/svi_lightning.py
imports: import argparse
  import lightning.pytorch
  import torch
  import pyro
  import pyro.distributions
  from pyro.infer import Trace_ELBO
  from pyro.infer.autoguide import AutoNormal
  from pyro.nn import PyroModule
label_desc: Use image node_id 3 for calling the PyroLightningModule obj's underlying member method code with example usage: obj.training_step(batch, batch_idx) and returns: loss