Training the RBM