摘要: 聚类loss的计算 要使得初始化的用户向量X在训练过程中得到优化,我们需要对前述示例进行一些修改,确保X是一个可训练的参数。在PyTorch中,这意味着我们需要将X定义为一个Parameter或者设置requires_grad=True。然而,由于X代表原始数据,通常我们不直接将其视为模型参数进行优 阅读全文
posted @ 2024-03-14 10:26 GraphL 阅读(32) 评论(0) 推荐(0)