Releases: iSiddharth20/Generative-AI-Based-Spatio-Temporal-Fusion
Releases · iSiddharth20/Generative-AI-Based-Spatio-Temporal-Fusion
Updated Math, Enhanced Functionalities
What is New in this Version?
- Flexibility to Add different independent Datasets for AutoEncoder and LSTM, now these components don't necessarily have to be trained on the same images.
- Ability to define independent and different batch size for AutoEncoder Training and LSTM Training.
- SSIM will now act as Regularization Term along with MSE in SSIMLoss Function.
- Updated and Corrected Math for MaxEnt Regularization Term.
- For LSTM Training, Each Individual Sequence will now be split to create Training/Validation Datasets.
Stable Multi-GPU Support
Core Functionality Achieved with PyTorch DDP.
Attached Zip File contains:
- Necessary Python Scripts
- requirements.txt
- Code Explanation in README.md
First Stable Release
Core Functionality Achieved.
Attached Zip File contains:
- Necessary Python Scripts
- requirements.txt
- Code Explanation in README.md