YiyanXu / DiffRec

Diffusion Recommender Model
168 stars 23 forks source link

betas out of range in gaussian_diffusion.py", line 35 #2

Open Guanyu-Lin opened 1 year ago

Guanyu-Lin commented 1 year ago

I am running "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.0005 0 1 log 1 0" for L-DiffRec, but It turns out to be with negative beta that out of range. I trace the code and found it uses "linear-var" as noise_schedule. I print the beta result and it is as: [ 5.00000000e-04, -6.25312656e-05, -6.25273557e-05, -6.25234463e-05, -6.25195374e-05]. Could you please help me to check the problem?

YiyanXu commented 1 year ago

You need to ensure the hyper-parameter args.noise_min < args.noise_max carefully.