use_gpu = False else: use_gpu = True os.environ["CUDA_VISIBLE_DEVICES"] = args.gpu device_list = args.gpu.split(",") for a in range(0, len(device_list)): gpu_list.append(int(a)) os.system("clear") config = create_config(configFilePath) cuda = torch.cuda.is_available() logger.info("CUDA available: %s" % str(cuda)) if not cuda and len(gpu_list) > 0: logger.error("CUDA is not available but specific gpu id") raise NotImplementedError parameters = init_all(config, gpu_list, args.checkpoint, "test", local_rank=args.local_rank) json.dump(test(parameters, config, gpu_list), open(args.result, "w", encoding="utf8"), ensure_ascii=False, sort_keys=True, indent=2)
args = parser.parse_args() configFilePath = args.config use_gpu = True gpu_list = [] if args.gpu is None: use_gpu = False else: use_gpu = True os.environ["CUDA_VISIBLE_DEVICES"] = args.gpu device_list = args.gpu.split(",") for a in range(0, len(device_list)): gpu_list.append(int(a)) os.system("clear") config = create_config(configFilePath) cuda = torch.cuda.is_available() logger.info("CUDA available: %s" % str(cuda)) if not cuda and len(gpu_list) > 0: logger.error("CUDA is not available but specific gpu id") raise NotImplementedError parameters = init_all(config, gpu_list, args.checkpoint, "test") json.dump(test(parameters, config, gpu_list), open(args.result, "w", encoding="utf8"), ensure_ascii=False, sort_keys=True, indent=2)
else: use_gpu = True os.environ["CUDA_VISIBLE_DEVICES"] = args.gpu device_list = args.gpu.split(",") for a in range(0, len(device_list)): gpu_list.append(int(a)) os.system("clear") config = create_config(configFilePath) cuda = torch.cuda.is_available() logger.info("CUDA available: %s" % str(cuda)) if not cuda and len(gpu_list) > 0: logger.error("CUDA is not available but specific gpu id") raise NotImplementedError # parameters = init_all(config, gpu_list, args.checkpoint, "test") parameters = [ init_all(config, gpu_list, args.checkpoint.replace('.pkl', 's.pkl'), "test"), init_all(config, gpu_list, args.checkpoint.replace('.pkl', 'm.pkl'), "test") ] json.dump(test(parameters, config, gpu_list), open(args.result, "w", encoding="utf8"), ensure_ascii=False, sort_keys=True, indent=2)