This is awesome! In fact it is so neat that really wanted to see it in action! Since I do not have 0.14 anymore, I decided to give it a try on 0.15, and it worked fine with some modifications: NiftyLogisticTrainNetwork was a bit of work: merged the original (v1.3.3) back into NiftyManiacs fork (to m...