wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

removed old template

Browse Source
This commit is contained in:
William Falcon
2019-07-26 21:39:53 -04:00
parent a5a80f35ec
commit 0ee0344820
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/examples/new_project_templates/lightning_module_template.py
-11
View File
@@ -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
# ---------------------