Skip to content

Latest commit

 

History

History
41 lines (29 loc) · 1.4 KB

README.md

File metadata and controls

41 lines (29 loc) · 1.4 KB

Collaborative Deep Reinforcement Learning

Collaborative Deep Reinforcement Learning (CDRL) is a collaborative framework to enable knowledge transfer among heterogeneous tasks.

Dependencies

Folder structure

|>CDRL

The code for experiments in paper section 5.4 Collaborative Deep Reinforcement Learning

|>heterogeneousTransfer

The code for experiments in paper section 5.3 Certificated Heterogeneous Transfer

|>model

The pre-trained model used for section heterogeneous transfer

How to run:

|>heterogeneousTransfer

python train.py --num-MTL-workers 8 0 --env-id PongDeterministic-v3_Bowling-v0

|>CDRL

python train.py --num-MTL-workers 8 8 --env-id PongDeterministic-v3_Bowling-v0

Preprint:

Collaborative Deep Reinforcement Learning
Kaixiang Lin, Shu Wang and Jiayu Zhou

Acknowledgement

This code refers to OpenAI universe starter agent.