@fadetoblack Yeah sure.
Here is my model and dataset file  https://gist.github.com/saahiluppal/e1ffeac4c6c6c3da045cf07d8f8df37e 6
And here is how i run it
model = ModelOne(config)
data = DataTypeOne(config)

trainer = pl.Trainer(gpus=1, max_epochs=config.epoch)
trainer.fit(model, data)