train.py 251 字节
Newer Older
T
tangwei12 已提交
1 2 3 4 5 6 7 8 9 10 11

import yaml
from .. trainer.factory import TrainerFactory

if __name__ == "__main__":

    with open('ctr-dnn_train.yaml', 'r') as rb:
        global_config = yaml.load(rb.read())

    trainer = TrainerFactory.craete(global_config)
    trainer.run()