What I did

What I learned

fill in later

class net(pl.LightningModule):
  def __init__(self, hparams):
    super(net, self).__init__()
        self.hparams = hparams
  def forward(self, x):
    pass
  def training_step(self, batch, batch_idx):
    pass
  def train_data_loader(self):
    pass
  def configure_optimizers(self):
    pass

You can also do it as a decorator, like this:

@pl.data_loader
def train_data_loader(self):
    pass

A decorator is python’s way of overriding an existing method in python.

What I will do next