wysiwyg icon indicating copy to clipboard operation
wysiwyg copied to clipboard

can't create the model with the config file

Open ConanCui opened this issue 3 years ago • 3 comments

Hi I run the training script with the config, and it shows that there is a typo in the config that "rpn" is spelled with "vpn" https://github.com/peiyunh/wysiwyg/blob/b1747a2aceb26bccfaeb825e0cac4b21f87eeb1e/second/configs/nuscenes/all.pp.mhead.vpn.config#L32

I correct it and run the train script with the config, and it still shows the error as below, could you tell that why would this happen?

Traceback (most recent call last): File "script.py", line 141, in <module> fire.Fire() File "/DATA3_DB7/data/kncui/miniconda3/envs/Lidar/lib/python3.6/site-packages/fire/core.py", line 138, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) File "/DATA3_DB7/data/kncui/miniconda3/envs/Lidar/lib/python3.6/site-packages/fire/core.py", line 468, in _Fire target=component.__name__) File "/DATA3_DB7/data/kncui/miniconda3/envs/Lidar/lib/python3.6/site-packages/fire/core.py", line 672, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) File "script.py", line 138, in test_nuscenes evaluate(cfg, model_dir, ckpt_path=ckpt_path, clean_after=False) File "/DATA3_DB7/data/kncui/wysiwyg/second/pytorch/train.py", line 479, in evaluate net = build_network(model_cfg, measure_time=measure_time).to(device) File "/DATA3_DB7/data/kncui/wysiwyg/second/pytorch/train.py", line 73, in build_network model_cfg, voxel_generator, target_assigner, measure_time=measure_time) File "/DATA3_DB7/data/kncui/wysiwyg/second/pytorch/builder/second_builder.py", line 142, in build direction_limit_offset=model_cfg.direction_limit_offset, File "/DATA3_DB7/data/kncui/wysiwyg/second/pytorch/models/net_multi_head.py", line 275, in __init__ super().__init__(*args, **kw) File "/DATA3_DB7/data/kncui/wysiwyg/second/pytorch/models/voxelnet.py", line 182, in __init__ num_direction_bins=self._num_direction_bins) File "/DATA3_DB7/data/kncui/wysiwyg/second/pytorch/models/rpn.py", line 242, in __init__ assert val == must_equal_list[0] AssertionError

ConanCui avatar Aug 09 '20 15:08 ConanCui