koulanurag / pfa

Policy Fusion Architecture (PFA): We investigate policy gradient approaches for reward decomposition in reinforcement Learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Policy Fusion Architecture - Pytorch

Installation

  • Install Pytorch [Version : 0.3.1] - (Platform dependent)
  • To install other requirements:
        pip install -r requirements.txt

Usage

  • If using render option: Start the visdom server before running the script
    python -m visdom.server
  • Policy Gradient:
    • Training: python main_pg.py --train
    • Testing: python main_pg.py --test
  • Hybrid Policy Gradient:
    • Training: python main_hpg.py --train
    • Testing: python main_hpg.py --test

About

Policy Fusion Architecture (PFA): We investigate policy gradient approaches for reward decomposition in reinforcement Learning


Languages

Language:Python 99.6%Language:Shell 0.4%