* Allowing validation dataset for computing validation loss

* added validation dataset

* fixing datasets errors

* Fixing linting errors