wuwenhai
/
bev-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #70 from JoeyforJoy/debug_jfj 

[debug] fix bug for benchmark
This commit is contained in:
Haotian (Ken) Tang 2022-07-12 09:28:26 -04:00 committed by GitHub
parent 36d34b73eb ae7f54a3e5
commit 98319eeaeb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tools/benchmark.py
View File

@ -7,7 +7,8 @@ from mmcv.runner import load_checkpoint, wrap_fp16_model
from mmdet3d.datasets import build_dataloader, build_dataset from mmdet3d.datasets import build_dataloader, build_dataset
from mmdet3d.models import build_fusion_model from mmdet3d.models import build_fusion_model
from torchpack.utils.config import configs
from mmdet3d.utils import recursive_eval
def parse_args(): def parse_args():
parser = argparse.ArgumentParser(description="MMDet benchmark a model") parser = argparse.ArgumentParser(description="MMDet benchmark a model")
@ -19,11 +20,11 @@ def parse_args():
args = parser.parse_args() args = parser.parse_args()
return args return args
def main(): def main():
args = parse_args() args = parse_args()
cfg = Config.fromfile(args.config) configs.load(args.config, recursive=True)
cfg = Config(recursive_eval(configs), filename=args.config)
# set cudnn_benchmark # set cudnn_benchmark
if cfg.get("cudnn_benchmark", False): if cfg.get("cudnn_benchmark", False):
torch.backends.cudnn.benchmark = True torch.backends.cudnn.benchmark = True