Mixing hparams and arguments in LightningModule.__init__() crashes load_from_checkpoint()
🐛 Bug
Right now, if you initialize a Lightning Module with a mixture of a Namespace (hparams) as well as additional arguments (say to a Dataset), load_from_checkpoint can't recover.
To Reproduce
Create a LightningModule as follows:
class Model(pl.LightningModule):
      def __init__(self, hparams, train_dataset, val_dataset):
              self.hparams = hparams
              self.train_dset, self.val_dset = train_dataset, val_dataset
              ...
Run training, then try to restore from checkpoint, via:
nn = Model.restore_from_checkpoint(<PATH>, train_dataset=None, val_dataset=None)
Expected behavior
Ideally, you'd just be able to pass in the additional arguments (as above) and everything would work.