Working on the epochs for training the RBM.