H-E-L-P / XID_plus

Next generation XID development

MIT License

3 stars 8 forks source link

Pyro backend VB #66

Open pdh21 opened 4 years ago

pdh21 commented 4 years ago

Expanding to more complicated models is going to require moving to Variational inference. Pyro looks like the best backend to use to be able to implement this

pdh21 commented 4 years ago

Standard SPIRE model test

Time comparison

No subsampling

fit_pyro=SPIRE.all_bands([prior250,prior350,prior500],n_steps=500,lr=0.001)

CPU times: user 2min 9s, sys: 20.6 s, total: 2min 29s Wall time: 1min 6s

with subsampling

fit_pyro=SPIRE.all_bands([prior250,prior350,prior500],n_steps=500,lr=0.001)

CPU times: user 1min 2s, sys: 7.24 s, total: 1min 9s Wall time: 37.7 s