YiyanXu / DiffRec

Diffusion Recommender Model
168 stars 23 forks source link

L-DiffRec betas out of range #18

Open woshiwangzihao opened 3 months ago

woshiwangzihao commented 3 months ago

ser num:108822iten num:94949data ready. running k-means on cuda:0.. [running kneans]: 0it [00:00,?it/s,center_shift=0.066783,iteration=1, tol=0 [running kneans]: 1it [00:00,10.89it/s,center_shift=0.002020,iteration=2,tol[running kneans]: 2it [00:00,15.43it/s, center_shift=0.000370,iteration=3, tol[running kneans]: 3it[00:00,23.13it/s,center_shift=0.000370, iteration=3, tol [running kneansj: 3it [00:00,23.13it/s, center_shift=0.000044,iteration=4,tol[running kneans]: 4it [00:00,25.34it/s, center_shift=0.000044,iteration=4,tol category length:[9495,85454] Latent dims of each category:[[30],[270]]Traceback (most recent call last): File "main.py" , line 133, in diffusion = gd.GaussianDiffusion(nean_type,args.noise_schedule, File "/media/wang/study/jhs/DiffRec-main/L-DiffRec/models/gaussian_diffusion y", line 35, in init assert (self. betas > 0).all() and (self. betas = 1).all(), "betas out of range" AssertionError: betas out of range

May I ask the author, when I reproduce L-Diffrec, according to the default parameter execution, there will be this error, I do not understand, please explain.

Andy3117006664 commented 2 months ago

我也有同样的问题,他的示例写错了“sh run.sh ml-1m_clean 5e-4 1e-4 0 0 400 2 [300] [] 0.05 [300] 10 x0 5 0.5 0.001 0.0005 0 1 log 1 0”的noise_min和noise_max要调换一下

YiyanXu commented 2 months ago

Sorry for the mistake in the run.sh script for L-Diffrec. The correct execution code should be: sh run.sh amazon-book_clean 5e-4 1e-4 0 0 400 2 [300] [] 0.05 [300] 10 x0 5 0.5 0.001 0.005 0 1 log 1 0 I have updated the repository with the corrected script.