Newbeeer / Poisson_flow

Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)
Apache License 2.0
846 stars 58 forks source link

How to set hyperparameter like gamma, sigma_end etc.? #9

Closed liuyezhou closed 1 year ago

liuyezhou commented 1 year ago

Hi, Could you please provide code of calculating average l2 norm on CIFAR-10? Thanks!

Newbeeer commented 1 year ago

Hi,

I calculated the l2 norm in another private project, and it's hard to elegantly integrate the code into current repo.

The calculation is pretty easy: (1) Setup a data loader for CIFAR-10; (2) Enumerate over the data points in CIFAR-10 to calculate the l2 norm.

Yilun

liuyezhou commented 1 year ago

Hi,

It's great job and thank your kindly reply!

I try to adopt your implementation to TTS/VC duty, and here is my part hyperparameter setting:

tau: 0.03 
data_norm: 9     # l2 normalization on hidden state by each frame
data_dim: 256    # hidden state channel

Could you give any advice about how to set other hyperparameter like gamma, sigma_end etc.?

Yezhou