def METHOD_NAME(self):
    print('Starting training test for SingleDemoGrasp...')
    self.learner.fit()
    old_weights = load_old_weights()
    new_weights = load_weights_from_file(os.path.join(self.learner.output_dir, "model_final.pth"))
    self.assertFalse(torch.equal(old_weights, new_weights),
                     msg="Fit method did not alter model weights")