2021年4月3日
模型蒸馏，蒸馏损失函数实现，蒸馏方法介绍
摘要： 1. 蒸馏损失函数代码实现 摘自：知识蒸馏loss求解方法 - tangjunjun - 博客园 (cnblogs.com) import torch import torch.nn as nn import numpy as np loss_f = nn.KLDivLoss() # 生成网络输出 阅读全文
posted @ 2021-04-03 19:44 Picassooo 阅读(626) 评论(0) 推荐(0) 编辑