TF-SegNet icon indicating copy to clipboard operation
TF-SegNet copied to clipboard

Session can be replaced by Estimator for training

Open akanimax opened this issue 6 years ago • 1 comments

The code organization is very good resulting in a clean and readable implementation of the SEGNET architecture. Good Job :+1:. However, the training loop is implemented by using the lower level Session api. This can be improved by using the new Estimator API.

akanimax avatar Mar 20 '18 04:03 akanimax

Thank you for the tip! I'll look into it, but I'm probably not gonna change the code yet as I'm currently working on an implementation of a new network. I'll be happy to accept a PR though if you have time :)

mathildor avatar Mar 20 '18 08:03 mathildor