Remove implementation for training demo