YOLOV8_Segment——ultralytics.ipynb 中 torchkeras梦中情炉接口 训练 报错：NotImplementedError: Module [YOLO] is missing the required "forward" function

[new-programer]

测试loss计算过程

for batch in dl_train: break

for key,value in batch.items(): if isinstance(value,torch.Tensor): batch[key] = batch[key].cuda()

model = model.cuda() model.train(data = 'balloon-seg.yaml') batch['img'] = batch['img'].float()/255.0 loss,_ = model.forward(batch) loss