Created by: yoptar
Estimator and NNModel instead of Trainable; __call__ on batch instead of infer on anything
Estimator
NNModel
Trainable
__call__
infer