'LEARNING_RATE': 1e-3,
'REPLAY_BUFFER_SIZE': 1000,
'MIN_REPLAY_BUFFER_SIZE': 100,
'EPSILON_START': 1,
'EPSILON_END': 0.1,
'EPSILON_DECAY_DURATION': 5000,
}
# Allow changing hyperparameters from command-line arguments
args = get_args(default_args=args_dict)
# Create wrapped environment
env = make_env(args.ENV_ID)
# Set Seed
set_seed(env, args.SEED)
# GPU or CPU
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
# Create agent
agent = Agent(env, device, args)
# Train agent for args.NB_FRAMES
agent.train()
# Save agent
agent.save()
# Test agent
agent.test(render=False)
#!/usr/bin/env python3
import torch
from agents import Agent
from commons import get_args, set_seed
from wrappers import make_env
# Get hyperparameters from arguments
args = get_args()
# Create wrapped environment
env = make_env(args.ENV_ID)
# Set Seed
set_seed(env, args.SEED)
# GPU or CPU
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
# Create agent
agent = Agent(env, device, args)
# Train agent for args.NB_FRAMES
agent.train()
# Save agent
agent.save()
# Test agent
agent.test()
