From 0ee034482049982d5b4fbc6be50cbca0c22c454e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 26 Jul 2019 21:39:53 -0400 Subject: [PATCH] removed old template --- .../lightning_module_template.py | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py index 0a4dab26..608e534e 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -154,17 +154,6 @@ class LightningTemplateModel(LightningModule): tqdm_dic = {'val_loss': val_loss_mean.item(), 'val_acc': val_acc_mean.item()} return tqdm_dic - # --------------------- - # MODEL SAVING - # --------------------- - def get_save_dict(self): - checkpoint = {'state_dict': self.state_dict()} - return checkpoint - - def load_model_specific(self, checkpoint): - self.load_state_dict(checkpoint['state_dict']) - pass - # --------------------- # TRAINING SETUP # ---------------------