KGAT_last-fm报错

最新推荐文章于 2022-08-01 16:56:59 发布

sinat_40147910

最新推荐文章于 2022-08-01 16:56:59 发布

阅读量284

点赞数

分类专栏：跑代码报错汇总文章标签： python

本文链接：https://blog.csdn.net/sinat_40147910/article/details/107289414

版权

在训练KGAT模型时，在last.fm数据集上遇到内部错误，具体表现为Dst tensor is not initialized。错误出现在训练的第139个epoch，尽管训练指标如recall、precision、hit和ndcg有所显示，但在保存权重并进行测试时发生了错误。问题可能与TensorFlow的设备分配和初始化有关。

摘要由CSDN通过智能技术生成

报错(139报错)

Epoch 119 [262.2s + 57.4s]: train==[65.18733=20.97127 + 43.56342 + 0.65262], recall=[0.08422, 0.16985], precision=[0.03384, 0.01735], hit=[0.35055, 0.60010], ndcg=[0.13051, 0.21696]
Epoch 129 [262.9s + 62.6s]: train==[60.66845=20.37745 + 39.62215 + 0.66882], recall=[0.08444, 0.17008], precision=[0.03393, 0.01735], hit=[0.35135, 0.60014], ndcg=[0.13133, 0.21763]
Epoch 139 [958.1s + 75.9s]: train==[55.55648=19.43076 + 35.44190 + 0.68381], recall=[0.08477, 0.17052], precision=[0.03402, 0.01733], hit=[0.35275, 0.59968], ndcg=[0.13134, 0.21732]
save the weights in path: weights/last-fm/kgat_si_sum_bi_l3/64-32-16/l0.0001_r1e-05-1e-05
Traceback (most recent call last):
File “/home/ubuntu/conda/conda3/envs/t