首先使用mmcls
1.在配置文件输入effientnet.py
loss_weight的时候 ,如下修改:
2.修改代码 mmdetection/mmdet/models/losses/cross_entropy_loss.py:
CrossEntropyLoss类,增加如下代码
if type(loss_weight) is list:
self.loss_weight = torch.from_numpy(np.array
首先使用mmcls
1.在配置文件输入effientnet.py
loss_weight的时候 ,如下修改:
2.修改代码 mmdetection/mmdet/models/losses/cross_entropy_loss.py:
CrossEntropyLoss类,增加如下代码
if type(loss_weight) is list:
self.loss_weight = torch.from_numpy(np.array