网站首页 站内搜索

搜索结果

查询Tags标签: similarities,共有 1条记录
  • SimCSE的loss实现-tensorflow2

    对比学习的核心就是loss的编写,记录下loss的tensorflow实现 def unsupervise_loss(y_pred, alpha=0.05):idxs = tf.range(y_pred.shape[0])y_true = idxs + 1 - idxs % 2 * 2y_pred = tf.math.l2_normalize(y_pred, dim = 1)similarities = tf.matmul(y_pred, y_pred,adj…

    2022/3/28 23:31:31 人评论 次浏览
扫一扫关注最新编程教程