Skip to content

Instantly share code, notes, and snippets.

@NMZivkovic
Last active December 23, 2019 11:00
Show Gist options
  • Save NMZivkovic/f07478940c7dbb4cf8be6ee6dbea3172 to your computer and use it in GitHub Desktop.
Save NMZivkovic/f07478940c7dbb4cf8be6ee6dbea3172 to your computer and use it in GitHub Desktop.
import base64
import imageio
import matplotlib
import matplotlib.pyplot as plt
import tensorflow as tf
from tf_agents.agents.dqn.dqn_agent import DqnAgent
from tf_agents.networks.q_network import QNetwork
from tf_agents.environments import suite_gym
from tf_agents.environments import tf_py_environment
from tf_agents.policies.random_tf_policy import RandomTFPolicy
from tf_agents.replay_buffers.tf_uniform_replay_buffer import TFUniformReplayBuffer
from tf_agents.trajectories import trajectory
from tf_agents.utils import common
# Globals
NUMBER_ITERATION = 20000
COLLECTION_STEPS = 1
BATCH_SIZE = 64
EVAL_EPISODES = 10
EVAL_INTERVAL = 1000
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment