Model Training Source Code

Hi,

I would like to retrain the CNN with different parameters  or do some minor changes in the network architecture. Would it be possible to also publish the source code for the model training?

Best,
Jens