5b0bdf006088f720bd21ae838e8903ee49282bde,run_application.py,,,#,9

Before Change

import utilities.misc_common as util
import utilities.parse_user_params as parse_user_params

if sys.version_info[0] >= 3:
    sys.stdout = os.fdopen(sys.stdout.fileno(), "w")
else:
    sys.stdout = os.fdopen(sys.stdout.fileno(), "w", 0)



class NetFactory(object):
    @staticmethod
    def create(name):
        if name == "highres3dnet":

After Change

            raise NotImplementedError


if __name__ == "__main__":
    args, csv_dict = parse_user_params.run()
    if util.has_bad_inputs(args):
        sys.exit(-1)
    if not (args.cuda_devices == """"):
        os.environ["CUDA_VISIBLE_DEVICES"] = args.cuda_devices
        print("set CUDA_VISIBLE_DEVICES env to {}".format(args.cuda_devices))
    net_class = NetFactory.create(args.net_name)

    if args.action == "train":
        import engine.training

        device_str = "gpu" if args.num_gpus > 0 else "cpu"
        engine.training.run(net_class, args, csv_dict, device_str)
    else:
        import engine.inference

        device_str = "gpu" if args.num_gpus > 0 else "cpu"
        engine.inference.run(net_class, args, csv_dict, device_str)

Instances