Very cool work! Wanted to see how much it has improved from previous methods but wondering what settings you are recommending. Did you use the default options or something else? I see you mention a few different flags in the readme which made me wonder.
The Weight --lambda_var=0.01
Compute density changing loss across images --var_all I have tested var_all=False.
Number of Flow Blocks --flow_blocks=1
Learning Rate of Flow --flow_lr=0.001
Different flows --flow_type=bnaf BNAF works best for me. Feel free to experiment other flows.
Very cool work! Wanted to see how much it has improved from previous methods but wondering what settings you are recommending. Did you use the default options or something else? I see you mention a few different flags in the readme which made me wonder.