407c5b32ce54f72fbc33537922f598f3df37409e,tools/train_shadownet.py,,init_args,#,27

Before Change

    parser.add_argument("-j", "--num_threads", type=int, default=int(os.cpu_count()/2),
                        help="Number of threads to use in batch shuffling")

    return parser.parse_args()


def train_shadownet(tfrecords_dir: str, charset_dir: str, model_dir: str, tboard_dir: str, cfg: EasyDict,
                    weights_path: str=None, decode: bool=False, num_threads: int=4):

After Change

    parser.add_argument("-j", "--num_threads", type=int, default=int(os.cpu_count()/2),
                        help="Number of threads to use in batch shuffling")

    args = parser.parse_args()

    config = load_config(args.config_file)
    if args.dataset_dir:
        config.cfg.PATH.TFRECORDS_DIR = args.dataset_dir
    if args.chardict_dir:
        config.cfg.PATH.CHAR_DICT_DIR = args.chardict_dir
    if args.model_dir:
        config.cfg.PATH.MODEL_SAVE_DIR = args.model_dir
    if args.tboard_dir:
        config.cfg.PATH.TBOARD_SAVE_DIR = args.tboard_dir

    return args, config.cfg


def train_shadownet(cfg: EasyDict, weights_path: str=None, decode: bool=False, num_threads: int=4):
    

Instances