Open jamiechoi1995 opened 1 year ago
when running to the switch epoch, I got the following error:
Traceback (most recent call last):
File "tools/train.py", line 115, in <module>
main()
File "tools/train.py", line 111, in main
runner.train()
File "/opt/conda/envs/mmdet3/lib/python3.8/site-packages/mmengine/runner/runner.py", line 1721, in train
model = self.train_loop.run() # type: ignore
File "/opt/conda/envs/mmdet3/lib/python3.8/site-packages/mmengine/runner/loops.py", line 96, in run
self.run_epoch()
File "/opt/conda/envs/mmdet3/lib/python3.8/site-packages/mmengine/runner/loops.py", line 109, in run_epoch
self.runner.call_hook('before_train_epoch')
File "/opt/conda/envs/mmdet3/lib/python3.8/site-packages/mmengine/runner/runner.py", line 1785, in call_hook
raise TypeError(f'{e} in {hook}') from None
TypeError: class `ClassBalancedDataset` in mmengine/dataset/dataset_wrapper.py: __init__() got an unexpected keyword argument 'pipeline' in <mmyolo.engine.hooks.yolox_mode_switch_hook.YOLOXModeSwitchHook object at 0x7f6dc78ddc10>
Prerequisite
🐞 Describe the bug
Here is my config:
Environment
sys.platform: linux Python: 3.8.16 | packaged by conda-forge | (default, Feb 1 2023, 16:01:55) [GCC 11.3.0] CUDA available: True numpy_random_seed: 2147483648 GPU 0,1: NVIDIA A100-PCIE-40GB CUDA_HOME: /usr/local/cuda NVCC: Cuda compilation tools, release 11.4, V11.4.152 GCC: gcc (Ubuntu 9.4.0-1ubuntu1~20.04.1) 9.4.0 PyTorch: 2.0.1+cu117 PyTorch compiling details: PyTorch built with:
TorchVision: 0.15.2+cu117 OpenCV: 4.7.0 MMEngine: 0.7.3 MMCV: 2.0.0 MMDetection: 3.0.0 MMYOLO: 0.5.0+
Additional information
No response