LCWALitModule

class LCWALitModule(dataset='nations', dataset_kwargs=None, mode=None, model='distmult', model_kwargs=None, batch_size=32, learning_rate=0.001, label_smoothing=0.0, optimizer=None, optimizer_kwargs=None)[source]

Bases: LitModule

A PyTorch Lightning module for training a model with LCWA training loop.

Create the lightning module.

Parameters: