Releases: valentingol/gan-facies-modeling
Releases · valentingol/gan-facies-modeling
v0.3.1
- 🚀 Speed up metric computation with JAX (CPU) (in function
utils.metrics.components.get_perimeter
) - ✨ Allow no truncation trick (default)
- 🐛 Fix bug in unit test
tests/utils/test_configs.py
that didn't restoresys.argv
(causing bug in some other tests)
v0.3.0
- ✨ Add 10/12 indicators from Rongier, in 2D and 3D + Wasserstein distances as metrics (in
utils/metrics/
) - ✨ Allow early stopping based on the absolute sum of the losses in training (which is now wandb log)
- ✨ Truncated input trick (in
utils/sagan/trainer.py
andapps/test.py
)
v0.2.3
- ✨ Allow attention layer at every layer in
utils/sagan/modules.py
- ✨ Allow more resolution: 32, 64, 128, 256 (further resolution can be added very easily)
- 🚸 Add coloured logs in
apps/train.py
v0.2.2
- ✨ Add seed to control more the randomness (
config.seed
) - ✨ Add possibility to start EMA for generator training after some steps (
configs.ema_start_step
) - ✨ Add custom initialization for
utils/sagan/modules.py
: orthogonal, normal and glorot (orthogonal especially is used by BigGAN architecture)
v0.2.1
- ✨ Add optional EMA for generator in
utils/sagan/trainer
- ✨ Add full_values parameter in
utils/sagan/modules
. If False, attention behaviour is similar to BigGAN (except that convolution layers have biaises) - 🚀 Speed trainer test (in
tests/utils/sagan/test_trainer.py
) and remove pre-trained networks that was only used for tests - ✅ Improve coverage to 99% in
utils
v0.2.0
- ✨ Add test script to visualize generated images from pre-trained generator in
apps/test.py
- ♻️ Refactor
apps/train.py
andutils/sagan/trainer.py
to improve clarity and pylint score - ✨ Add general training utilities in
utils/train
v0.1.0
First release 🎉
- 📝 Add complete readme
- 🍱 Add third party license (GANSim) and 📝 mention source repository for sagan code
v0.0.2
- 🐛 Fix configs merging with wandb sweep in
apps/train.py
- 🚀 Pylint workflow no longer needs to install requirements.txt
- 🐛 Fix type hint list[...] -> List[...] to be compatible with Python3.8+ in
utils/sagan/modules.py
- ✨ Add
tests/basic_checks.sh
to run some checks before push/PR
v0.0.1
🎉 Initial commit
- Unconditional SAGAN
- Wandb integration