added interface for train_continuous