Update README.md

README.md

@@ -27,7 +27,7 @@ Training progress:
 ![training](training_graph.jpg)
 
 Numbers on X axis are average over 40 episodes, each lasting for about 500 timesteps on average. So in total the agent was trained over about 5e6 timesteps.
-Learning rate decay schedule: <code>torch.optim.lr_scheduler.StepLR(opt, step_size=4000, gamma=0.7)</code>
+Learning rate decay schedule: <code>torch.optim.lr_scheduler.StepLR(opt, step_size=4000, gamma=0.7)</code> Training code is shown in the training.py file for reference.
 
 Minimal code to use the agent:</br>
 ```