As a first step I would like to replicate the keras loss and validation loss printed values after each epoch using tensorflow. My current code looks like this:
