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)
