add rbm pretraining