removed dead code in model save

This commit is contained in:
William Falcon 2019-07-24 15:44:04 -04:00
parent b836e6f321
commit b7ca857434
1 changed files with 29 additions and 0 deletions

View File

@ -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