def __init__(self, project_config):
config = base_config(project_config)
self.project = project_config
self.url = config.get('jms_host', special_select='base')
self.username = config.get('jms_user', special_select='base')
self.password = config.get('jms_pass', special_select='base')
self.headers = {
'Content-Type': 'application/json',
'accept': 'application/json'
}
def assents(project_config):
from config import base_config
cf = base_config(project_config)
if cf.get('platform_type') == 'aliyun':
ali_assent = ali_ecs_assent(ali_access_key=cf.get('access_key'),
ali_secret_key=cf.get('secret_key'),
ali_region=cf.get('region'))
now_assents = ali_assent.get_aliyun_hosts(
register_prefix=cf.get('register_prefix'),
default_re=cf.get('default_regular', special_select='base'),
extend_re=cf.get('extend_regular'),
project=project_config)
elif cf.get('platform_type') == 'aws':
massent = aws_ec2_assent(aws_access_key=cf.get('access_key'),
aws_secret_key=cf.get('secret_key'),
aws_region=cf.get('region'))
now_assents = massent.get_assents(
register_prefix=cf.get('register_prefix'),
default_re=cf.get('default_regular', special_select='base'),
extend_re=cf.get('extend_regular'),
project=project_config)
return now_assents, cf.get('jms_assents_group'), cf.get(
'jms_domain_name'), cf.get('jms_admin_user')
def __init__(self, project_config):
config = base_config(project_config)
self.etcd_host = config.get('ETCD_HOST')
self.etcd_root = config.get('ETCD_ROOT')
"""
import sys
sys.dont_write_bytecode = True
from config import base_config
from optimizer import get_optimizer
from loss import get_loss_function
from model import get_model
from metrics import get_metrics_lst
from callback import get_callbacks
from trainer import Trainer
from data import get_datasets
if __name__ == "__main__":
config = base_config()
config.METRICS_LST = get_metrics_lst()
config.OPTIMIZER = get_optimizer()
config.LOSS_FUNC = get_loss_function()
config.CALLBACK_LST = get_callbacks(config)
config.display()
model = get_model(config)
datasets = get_datasets(config)
trainer = Trainer(datasets, model, config)
trainer._compile()
trainer.train()