added safeguards for callbacks in loading saving

This commit is contained in:
William Falcon
2019-07-24 11:45:59 -04:00
parent aac5ba00ef
commit 926fa206ff
+10 -10
View File
@@ -1,7 +1,7 @@
import pytest
from pytorch_lightning import Trainer
from pytorch_lightning.examples.new_project_templates.lightning_module_template import LightningTemplateModel
from test_tube.argparse_hopt import TTNamespace
from argparse import Namespace
from test_tube import Experiment
from pytorch_lightning.callbacks import ModelCheckpoint
import numpy as np
@@ -15,14 +15,14 @@ import pdb
def get_model():
# set up model with these hyperparams
root_dir = os.path.dirname(os.path.realpath(__file__))
hparams = TTNamespace(**{'drop_prob': 0.2,
'batch_size': 32,
'in_features': 28*28,
'learning_rate': 0.001*8,
'optimizer_name': 'adam',
'data_root': os.path.join(root_dir, 'mnist'),
'out_features': 10,
'hidden_dim': 1000})
hparams = Namespace(**{'drop_prob': 0.2,
'batch_size': 32,
'in_features': 28*28,
'learning_rate': 0.001*8,
'optimizer_name': 'adam',
'data_root': os.path.join(root_dir, 'mnist'),
'out_features': 10,
'hidden_dim': 1000})
model = LightningTemplateModel(hparams)
return model, hparams
@@ -61,7 +61,7 @@ def main():
# exp file to get meta
exp = get_exp(False)
exp.add_meta_from_hyperopt(hparams)
exp.argparse(hparams)
exp.save()
# exp file to get weights