From b7ca8574347eeb56e83811f9be5d1fa55ed68ef1 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 15:44:04 -0400 Subject: [PATCH] removed dead code in model save --- pytorch_lightning/root_module/model_saving.py | 29 +++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/pytorch_lightning/root_module/model_saving.py b/pytorch_lightning/root_module/model_saving.py index 0569c89022..818c194748 100644 --- a/pytorch_lightning/root_module/model_saving.py +++ b/pytorch_lightning/root_module/model_saving.py @@ -184,3 +184,32 @@ class TrainerIO(object): return max(ckpt_vs) + +def load_hparams_from_tags_csv(tags_csv): + from argparse import Namespace + import pandas as pd + + tags_df = pd.read_csv(tags_csv) + dic = tags_df.to_dict(orient='records') + + ns_dict = {row['key']: convert(row['value']) for row in dic} + + ns = Namespace(**ns_dict) + return ns + + +def convert(val): + constructors = [int, float, str] + + if type(val) is str: + if val.lower() == 'true': + return True + if val.lower() == 'false': + return False + + for c in constructors: + try: + return c(val) + except ValueError: + pass + return val