@ptrblck I’ve submitted a pull request with updates to the reinforcement_q_learning.py tutorial. I’ve made the DQN network accept the number of outputs and updated the example to obtain the number of actions from the gym environment action space. This will help avoid similar issues for others who my try the DQN example with different gym environments.

  
      github.com/pytorch/tutorials
  
  
      
    
  


  Fixes for PyTorch reinforcement learning DQN tutorial. 5



  by edowson
  on 06:12PM - 25 Mar 19 UTC



  2 commits
  changed 2 files
  with 13 additions
  and 6 deletions.