Fix a number of issues with the infrastructure, no major rework
Fix a number of issues present in the code:
- type issues
- README.md issues
- variable initializations
- model issues
- training loss / number of batches calculation issues
Fix a number of issues present in the code: