|
- import os
- import argparse
- # project_root = os.path.abspath(
- # os.path.dirname(os.path.realpath(__file__)) + os.path.sep + "..")
- # print('project_root:', project_root)
- # EXEC_PATH = os.path.join(project_root, 'mindspore-client')
-
- parser = argparse.ArgumentParser(description='Image classification')
- parser.add_argument('--uuid', type=str, default=None, help='UUID')
- parser.add_argument('--data_path', type=str, default=None, help='DATA_PATH')
- parser.add_argument('--thgy_host', type=str, default=None, help='THGY_HOST')
- parser.add_argument('--thgy_port', type=str, default=None, help='THGY_POST')
- parser.add_argument('--agent_host', type=str, default=None, help='AGENT_HOST')
- parser.add_argument('--agent_port', type=str, default=None, help='AGENT_POST')
- parser.add_argument('--train_url', type=str, default=None, help='train_url')
- parser.add_argument('--data_url', type=str, default=None, help='data_url')
- parser.add_argument('--num_epoch', type=int, default=80, help='global step')
- parser.add_argument('--globalStept', type=int, default=1, help='global epochs')
- parser.add_argument('--dataset_path', type=str, default=None, help='Dataset path')
- parser.add_argument('--dataset', type=str, default=None, help='Dataset, either cifar10 or imagenet2012')
- parser.add_argument('--net', type=str, default=None, help='Resnet Model, either resnet18 or resnet50 or resnet101')
- parser.add_argument('--is_client', type=int, default=0, help='is Client')
- args_opt = parser.parse_args()
- print(args_opt)
-
- # if args_opt.is_client==1:
- # print('@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@')
- # os.system('bash /home/work/user-job-dir/mindspore-client/start_bp.sh {} {} {} {} {} {} {} {} {}'.format(args_opt.dataset_path, args_opt.uuid, args_opt.thgy_host, args_opt.thgy_port, args_opt.agent_host, args_opt.agent_port, args_opt.train_url, args_opt.data_url, args_opt.globalStept))
- # else:
- # os.system('bash /home/work/user-job-dir/mindspore-client/start.sh {} {} {} {} {} {} {} {}'.format(args_opt.train_url, args_opt.data_url, args_opt.net, args_opt.dataset, args_opt.dataset_path, args_opt.globalStept, args_opt.uuid, args_opt.num_epoch))
-
-
- if args_opt.is_client==1:
- print('@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@')
- os.system('bash start_bp.sh {} {} {} {} {} {} {} {} {}'.format(args_opt.dataset_path, args_opt.uuid, args_opt.thgy_host, args_opt.thgy_port, args_opt.agent_host, args_opt.agent_port, args_opt.train_url, args_opt.data_url, args_opt.globalStept))
- else:
- os.system('bash start.sh {} {} {} {} {} {} {} {}'.format(args_opt.train_url, args_opt.data_url, args_opt.net, args_opt.dataset, args_opt.dataset_path, args_opt.globalStept, args_opt.uuid, args_opt.num_epoch))
-
- # os.system('nohup ./client py -c "python hello.py" --uuid={}'.format(args_opt.uuid))
|