File size: 889 Bytes
25986db | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | import argparse
import os
import _init_paths
from lib.train.admin import create_default_local_file_ITP_train
from lib.test.evaluation import create_default_local_file_ITP_test
def parse_args():
parser = argparse.ArgumentParser(description='Create default local file on ITP or PAI')
parser.add_argument("--workspace_dir", type=str, required=True) # workspace dir
parser.add_argument("--data_dir", type=str, required=True)
parser.add_argument("--save_dir", type=str, required=True)
args = parser.parse_args()
return args
if __name__ == "__main__":
args = parse_args()
workspace_dir = os.path.realpath(args.workspace_dir)
data_dir = os.path.realpath(args.data_dir)
save_dir = os.path.realpath(args.save_dir)
create_default_local_file_ITP_train(workspace_dir, data_dir)
create_default_local_file_ITP_test(workspace_dir, data_dir, save_dir)
|