training and inference should be same thing

that will gain real time improving model