Problems to change the .pt

[DerBra1n]

If I use the existing .pt then the bot works fine, but as soon as I insert my best.pt I get an error.
I use 4 NC instead of 2 like you, what do I have to change so that the Aimbot recognizes my best.pt?
Here the error if I want to exchange the .pt:
experimental.py, Line 80, ckpt = torch.load(attempt_download(w))
Error:
Exception has occurred: AttributeError
Can't get attribute 'DetectionModel' on <module 'models.yolo' from 'd:\Downloads\Yolov5ForCSGO-main\Main\models\yolo.py'>
File "D:\Downloads\Yolov5ForCSGO-main\Main\experimental.py", line 80, in attempt_load
ckpt = torch.load(attempt_download(w))
File "D:\Downloads\Yolov5ForCSGO-main\Main\cs_model.py", line 18, in load_model
model = attempt_load(weights, device=device)
File "D:\Downloads\Yolov5ForCSGO-main\Main\main.py", line 34, in
model = load_model()
AttributeError: Can't get attribute 'DetectionModel' on <module 'models.yolo' from 'd:\Downloads\Yolov5ForCSGO-main\Main\models\yolo.py'>

I hope anybody can help me :)

[Lucid1ty]

@DerBra1n
The error message you're seeing is related to the PyTorch model loading process. It seems that the script is unable to find the DetectionModel class in the models.yolo module. This could be due to a few reasons:

1. Mismatched versions: The model file best.pt might have been trained with a different version of the YOLOv5 codebase. If the DetectionModel class or its attributes have been modified in the version you're using, it could lead to this error.

2. Incorrect directory structure: Ensure that the directory structure of your project matches that of the original YOLOv5 repository. If files have been moved around, it could cause issues with loading the model.

3. Changes in the number of classes: You mentioned that you're using 4 classes instead of 2. If best.pt was trained with a different number of classes, you'll need to ensure that your .yaml file and any other relevant configuration files reflect this change.

To resolve this issue, you could try the following steps:

• Check out the exact commit of the YOLOv5 repository that was used to train best.pt, and use that version of the codebase.
• Double-check your directory structure and ensure all files are in their correct locations.
• Update your .yaml file and any other configuration files(especially the targeting part of the code) to correctly specify that you're working with 4 classes.