Danisch

Policy Gradient Issue: ValueError: Shapes (20, 1) and (20, 2) are incompatible

1

Hi. The code [Code](https://github.com/adventuresinML/adventures-in-ml-code/blob/master/policy_gradient_reinforce_tf2.py ) is not working with this line: `loss = network.train_on_batch(states, discounted_rewards)`.