KaihongLi
KaihongLi
> when I was trying to run the model in my custom weights.I'm getting this error > > ``` > > File "C:\Users\acer\.spyder-py3\metallic surface defect detection\3_onnx_cpu_detec.py", line 31, in detect...
The training batch size is 1, but each batch contains 5 images for training.
PyTorch version: 2.3.1 CUDA version: 12.1 run test_net.py with config_file vid_R_50_C4_MEGA_1x.yaml model_weight MEGA_R_50.pth 