diff --git a/pipeline.py b/pipeline.py index 58b5380..5442935 100644 --- a/pipeline.py +++ b/pipeline.py @@ -130,8 +130,8 @@ class Components : logger = factory.instance(type='mongo.MongoWriter',args={'dbname':'aou','doc':args['context']}) log_folder = args['logs'] if 'logs' in args else 'logs' partition = args['partition'] if 'partition' in args else '' - log_folder = os.sep.join([log_folder,args['context'],partition]) - + log_folder = os.sep.join([log_folder,args['context'],str(partition)]) + _args = {"batch_size":10000,"logs":log_folder,"context":args['context'],"max_epochs":150,"column":args['columns'],"id":"person_id","logger":logger} _args['max_epochs'] = 150 if 'max_epochs' not in args else int(args['max_epochs']) _args['num_gpu'] = int(args['num_gpu']) if 'num_gpu' in args else 1