#! /usr/bin/python
# -*- coding: utf-8 -*-

import argparse

from tensorlayer.cli import train

if __name__ == "__main__":
    parser = argparse.ArgumentParser(prog='tl')
    subparsers = parser.add_subparsers(dest='cmd')
    train_parser = subparsers.add_parser('train', help='train a model using multiple local GPUs or CPUs.')
    train.build_arg_parser(train_parser)
    args = parser.parse_args()
    if args.cmd == 'train':
        train.main(args)
    else:
        parser.print_help()