bugfix for rbmdnn train