santanamartinir / utn-deeplearning

Assignment: Generative HPO with Diffusers
0 stars 0 forks source link

Generate plots #5

Open suman1209 opened 1 month ago

suman1209 commented 1 month ago

Task Details

the task is to generate the performance plots after training the diffusion model

TODOs

suman1209 commented 2 weeks ago

You can now generate plots by running main.py in the 'main' branch, Please adjust the timesteps in the inference and the epochs to play around and experiment with results. I noticed that it takes a very long time for sampling / recommendations when you use T = 1000

santanamartinir commented 2 weeks ago

I created a plotting.py to compare against all three baselines using the .json inside /results

suman1209 commented 2 weeks ago

There are path related things that need to be checked when running in your local pc, but running main.py should give you some results like the one below image