-
Notifications
You must be signed in to change notification settings - Fork 6
Home
XT_Yang edited this page Mar 4, 2021
·
10 revisions
Welcome to the DRL_Implementation wiki!
This tutorial will walk you through the usages, building blocks and programing logic of this Pytorch-based Deep Reinforcement Learning implementation package.
- Using off-the-shell agent.
- What's in the package?
After the tutorial, you can proceed to create your own agent using common components of modern DRL algorithms available from this package. Feel free to raise issues. Here are some tips that may help:
- It is recommended to start by modifying an existing agent.
- For compatibility, make sure your networks and statistics are saved in dictionaries.
Here's a starting example:
import time
import numpy as np
import torch as T
import torch.nn.functional as F
from torch.optim.adam import Adam
from drl_implementation.agent.agent_base import Agent
from drl_implementation.agent.utils.networks_mlp import Actor, Critic
from drl_implementation.agent.utils.exploration_strategy import GaussianNoise
class MyDDPG(Agent):
def __init__(self, ...):
...