Adding conditional attention
This MR adds the following:
- Adding new train script for DIPS Conditional Attention + example train config.
- Adding support for new DIPS Conditional Attention model in the validation/evaluation chain.
- Adding integration test for train/validation/evaluation of DIPS Conditional Attention.
- Fixing some issues in the
tf_tools/modelswith the masking (also adapted the unit test). - Adding new generator for DIPS Conditional Attention.
- Adding some flexibility to the loading of the
loading_validation_datafunctions. - Adding compatibility of the
evaluate_model.pyfor DIPS and DIPS Conditional Attention. - Adding DIPS Conditional Attention to possible models for
plotting_epoch_performance.py - Adding DIPS Conditional Attention train script to the
setup.py. - Make the calculation of the Saliency maps steerable in the train config of the DIPS models.
Edited by Alexander Froch