I had tried this
model = PL_MODULE(hidden_size = 50, learning_rate = 0.0001, num_layers = 5, batch_size = 128)   
del model

if this is what you are referring to but it didn’t work. I could not understand your comment on loading the data. Is it inefficient to do this way? Because it doesn’t crash outside of the lightning module.