this program enables torch model keras-like train and outputs - Taku-ichimura/keras_like_model