JoeArmani
training and inference updates
5b413d1