How to train the model and which kind of GPU for training?

[happsky]

Thanks for sharing this amazing code, but I have a question that how to train the model and which kind of GPU for training? I only find predict.py, could you mind providing train.py for me?

[XGBoost]

Thank for you sharing the code, I want to explore the process of the train step, so could you mind providing the train step code and the corresponding description?

[MaybeShewill-CV]

@XGBoost You are welcomed to test my tensorflow version here https://github.com/MaybeShewill-CV/attentive-gan-derainnet. By the way I am also looking forward to the training procedure of this amazing work from the origin author.:)