focal loss可以用在VGG16中吗？或者是mobile net中？
我想用focal loss代替原模型的损失函数，基于pytorch的。在pytorch中，损失函数定义很简单：
criterion = torch.nn.CrossEntropyLoss()
怎么把它替换成focal loss呢？怎么写呢？
接着就是下面这段代码了
我想用focal loss代替原模型的损失函数，基于pytorch的。在pytorch中，损失函数定义很简单：
criterion = torch.nn.CrossEntropyLoss()
怎么把它替换成focal loss呢？怎么写呢？