Add ddpm model

[marianovitasari20]

Closes #55

[github-actions]

Coverage report

Click to see where and how coverage changed

File	Statements	Missing	Coverage	Coverage (new stmts)	Lines missing
ice_station_zebra/models/common
__init__.py
timeembed.py					26-30, 37
ice_station_zebra/models/diffusion
__init__.py
gaussian_diffusion.py					41-74, 93-98, 117-137, 153-156, 177-182
unet_diffusion.py					49-102, 123-173, 189-204, 217-220
ice_station_zebra/models/processors
__init__.py
ddpm.py					24-28, 40-46, 64-80
ice_station_zebra/types
__init__.py
complex_datatypes.py					37, 52-59, 63-65, 69
enums.py
simple_datatypes.py
typedefs.py
Project Total

This report was generated by python-coverage-comment-action

[jemrobinson]

I can review this today/tomorrow.

[jemrobinson]

We should use B, C, H, W format throughout the diffusion process. Happy to help with implementing this if it's not clear how to do it.

We also don't need EMA to be implemented inside the model, as we're running EMA in a lightning callback.

[jemrobinson]

Also, can you add some tests for the model? If you merge main into this branch, you can see some examples in tests/models/test_processors.py.

[marianovitasari20]

Also, can you add some tests for the model? If you merge main into this branch, you can see some examples in tests/models/test_processors.py.

#83

[marianovitasari20]

Thank you so much, @jemrobinson, for the useful feedback. I will adjust them soon.

[jemrobinson]

LGTM