DL Interface improved

DL interface for training has been improved and resembles more pytorch-lightning-ish API. Thus, improving readability and eas-of-use for users

Check below links for more details: