Hi Burhan,
There is an issue when I run your code. It seems that yolov8 has modified some interfaces.
Traceback (most recent call last):
File "c:\Users\16119\Desktop\ClassifyDefectMap_MixedWM38-master\yolo_wafer_classification.py", line 13, in
model.trainer.build_dataset(img_path=data_dir) # loads data, but misses 'val_dir'
AttributeError: 'NoneType' object has no attribute 'build_dataset'