Validation loss is much lower than training loss. There is probably something missing in the divisions before being printed. Need to check this out.