eguoguo321
Results
4
issues of
eguoguo321
here is my code `model = create_model_test(args, load_head=True).cuda() ########### eliminate BN for faster inference ########### model.load_state_dict(torch.load(args.model_path, map_location='cpu')) model = model.cpu() model = InplacABN_to_ABN(model) model = fuse_bn_recursively(model) model = model.cpu() `...