update of RBM training function, added a probe