NameError: name 'Mish' is not defined

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4

1.87k stars 585 forks source link

Open XFR1998 opened 3 years ago

XFR1998 commented 3 years ago

为什么我训练yolov4.weights时出现这个，应该如何解决

WongKinYiu commented 3 years ago

install mish-cuda

xinhaojin commented 1 year ago

import torch.nn as nn nn.Mish()