[X] I have searched the YOLOAir issues and found no similar bug report.
YOLOAir Component
No response
Bug
Fusing layers...
RepConv.fuse_repvgg_block
RepConv.fuse_repvgg_block
RepConv.fuse_repvgg_block
IDetect.fuse
/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/functional.py:478: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:2894.)
return _VF.meshgrid(tensors, kwargs) # type: ignore[attr-defined]
Model Summary: 321 layers, 36490750 parameters, 6194944 gradients
Traceback (most recent call last):
File "val.py", line 396, in
main(opt)
File "val.py", line 369, in main
run(vars(opt))
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
return func(*args, kwargs)
File "val.py", line 166, in run
model.warmup(imgsz=(1 if pt else batch_size, 3, imgsz, imgsz), half=half) # warmup
File "/usr/pyCode/yoloair/models/common.py", line 483, in warmup
self.forward(im) # warmup
File "/usr/pyCode/yoloair/models/common.py", line 422, in forward
y = self.model(im) if self.jit else self.model(im, augment=augment, visualize=visualize)
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, *kwargs)
File "/usr/pyCode/yoloair/models/yolo.py", line 172, in forward
return self._forward_once(x, profile, visualize) # single-scale inference, train
File "/usr/pyCode/yoloair/models/yolo.py", line 195, in _forward_once
x = m(x) # run
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(input, kwargs)
File "/usr/pyCode/yoloair/models/Detect/MuitlHead.py", line 211, in fuseforward
elif self.end2end:
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1207, in getattr
raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'IDetect' object has no attribute 'end2end'
Search before asking
YOLOAir Component
No response
Bug
Fusing layers... RepConv.fuse_repvgg_block RepConv.fuse_repvgg_block RepConv.fuse_repvgg_block IDetect.fuse /home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/functional.py:478: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:2894.) return _VF.meshgrid(tensors, kwargs) # type: ignore[attr-defined] Model Summary: 321 layers, 36490750 parameters, 6194944 gradients Traceback (most recent call last): File "val.py", line 396, in
main(opt)
File "val.py", line 369, in main
run( vars(opt))
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
return func(*args, kwargs)
File "val.py", line 166, in run
model.warmup(imgsz=(1 if pt else batch_size, 3, imgsz, imgsz), half=half) # warmup
File "/usr/pyCode/yoloair/models/common.py", line 483, in warmup
self.forward(im) # warmup
File "/usr/pyCode/yoloair/models/common.py", line 422, in forward
y = self.model(im) if self.jit else self.model(im, augment=augment, visualize=visualize)
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, *kwargs)
File "/usr/pyCode/yoloair/models/yolo.py", line 172, in forward
return self._forward_once(x, profile, visualize) # single-scale inference, train
File "/usr/pyCode/yoloair/models/yolo.py", line 195, in _forward_once
x = m(x) # run
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(input, kwargs)
File "/usr/pyCode/yoloair/models/Detect/MuitlHead.py", line 211, in fuseforward
elif self.end2end:
File "/home/lyd/anaconda3/envs/lyd/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1207, in getattr
raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'IDetect' object has no attribute 'end2end'
Environment
No response
Minimal Reproducible Example
No response
Additional
No response
Are you willing to submit a PR?