Merlin二塔模型数据训练错误（ValueError: high is out ofbounds for int32）

Question

model.compile（optimizer =“adam”，run_eagerly = False，metrics = [mm.RecallAt（10），mm.NDCGAt（10）]） model.fit（训练，validation_data=valid，batch_size=4096，epochs=3）

ValueError                                Traceback (most recent call last)

C:\ProgramData\Anaconda3\lib\site-packages\merlin\models\loader\backend.py in _shuffle_indices(self)
    326     @annotate("_shuffle_indices", color="darkgreen", domain="nvt_python")
    327     def _shuffle_indices(self):
--> 328         generate_local_seed(self.global_rank, self.global_size)
    329         if self.seed_fn:
    330             new_seed = self.seed_fn()

C:\ProgramData\Anaconda3\lib\site-packages\merlin\core\dispatch.py in generate_local_seed(global_rank, global_size)
    647         cp.random.seed(seeds[global_rank].get())
    648     else:
--> 649         seeds = random_state.randint(0, 2**32, size=global_size)
    650     return seeds[global_rank]
    651 

mtrand.pyx in numpy.random.mtrand.RandomState.randint()

_bounded_integers.pyx in numpy.random._bounded_integers._rand_int32()

ValueError: high is out of bounds for int32

我尝试使用 Nvidia Merlin 训练模型

期待模型迭代步骤。

Answer 1

您好，这里是对我有用的解决方案：

转到您的文件夹 AppData\Roaming\Python\Python_version_here\site-packages\merli