def main(args=None):
# Parse arguments
if args is None:
args = sys.argv[1:]
args = parse_args(args)
# Check if a GPU ID was set
if args.gpu:
os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu
set_session(get_session())
summary_writer = tf.summary.FileWriter(args.type + "/tensorboard_" +
args.env)
# Environment Initialization
# if(args.is_atari):
# # Atari Environment Wrapper
# env = AtariEnvironment(args)
# state_dim = env.get_state_size()
# action_dim = env.get_action_size()
# else:
# Standard Environments
env = Environment(gym.make(args.env), args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_dim = gym.make(args.env).action_space.n
# Pick algorithm to train
# if(args.type=="DDQN"):
algo = DDQN(action_dim, state_dim, args)
# elif(args.type=="A2C"):
# algo = A2C(action_dim, state_dim, args.consecutive_frames)
# elif(args.type=="A3C"):
# algo = A3C(action_dim, state_dim, args.consecutive_frames, is_atari=args.is_atari)
# elif(args.type=="DDPG"):
# algo = DDPG(action_dim, state_dim, act_range, args.consecutive_frames)
# if args.actor_path is not None and args.critic_path is not None:
# print("Loading weights...")
# algo.load_weights(args.actor_path, args.critic_path)
# Train
stats = algo.train(env, args, summary_writer)
# Export results to CSV
if (args.gather_stats):
df = pd.DataFrame(np.array(stats))
df.to_csv(args.type + "/logs.csv",
header=['Episode', 'Mean', 'Stddev'],
float_format='%10.5f')
# Save weights and close environments
exp_dir = '{}/models/'.format(args.type)
if not os.path.exists(exp_dir):
os.makedirs(exp_dir)
export_path = '{}{}_ENV_{}_NB_EP_{}_BS_{}'.format(exp_dir, args.type,
args.env,
args.nb_episodes,
args.batch_size)
algo.save_weights(export_path)
env.env.close()
def main(args=None):
# Parse arguments
if args is None:
args = sys.argv[1:]
args = parse_args(args)
if args.wandb:
wandb.init(entity=args.wandb_id, project=args.wandb_project)
# Environment Initialization
if (args.is_atari):
# Atari Environment Wrapper
env = AtariEnvironment(args)
state_dim = env.get_state_size()
action_dim = env.get_action_size()
elif (args.type == "DDPG"):
# Continuous Environments Wrapper
env = Environment(gym.make(args.env), args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_space = gym.make(args.env).action_space
action_dim = action_space.high.shape[0]
act_range = action_space.high
else:
# Standard Environments
env = Environment(gym.make(args.env), args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_dim = gym.make(args.env).action_space.n
# Pick algorithm to train
if (args.type == "DDQN"):
algo = DDQN(action_dim, state_dim, args)
elif (args.type == "A2C"):
algo = A2C(action_dim, state_dim, args.consecutive_frames)
elif (args.type == "A3C"):
algo = A3C(action_dim,
state_dim,
args.consecutive_frames,
is_atari=args.is_atari)
elif (args.type == "DDPG"):
algo = DDPG(action_dim, state_dim, act_range, args.consecutive_frames)
# Train
stats = algo.train(env, args) # e, mean, stdev: e is episode
# Export results to CSV
if (args.gather_stats):
df = pd.DataFrame(np.array(stats))
df.to_csv(args.type + "/logs.csv",
header=['Episode', 'Mean', 'Stddev'],
float_format='%10.5f')
# Save weights and close environments
exp_dir = '{}/models/'.format(args.type)
if not os.path.exists(exp_dir):
os.makedirs(exp_dir)
export_path = '{}{}_ENV_{}_NB_EP_{}_BS_{}'.format(exp_dir, args.type,
args.env,
args.nb_episodes,
args.batch_size)
algo.save_weights(export_path)
env.env.close()
def main(args=None):
# Parse arguments
if args is None:
args = sys.argv[1:]
args = parse_args(args)
# Check if a GPU ID was set
# if args.gpu:
# os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu
# sess = get_session()
# set_session(sess)
# K.set_session(sess)
# with tf.device('/gpu:0'):
# config = tf.ConfigProto()
# config.gpu_options.allow_growth = True
# sess = tf.Session(config=config)
# K.set_session(sess)
# set_session(sess)
summary_writer = tf.summary.FileWriter(args.type + "/tensorboard_" +
args.env)
# Environment Initialization
if (args.is_atari):
# Atari Environment Wrapper
env = AtariEnvironment(args)
state_dim = env.get_state_size()
action_dim = env.get_action_size()
elif (args.type == "DDPG"):
# Continuous Environments Wrapper
# env = Environment(gym.make(args.env), args.consecutive_frames)
# env.reset()
# state_dim = env.get_state_size()
# action_space = gym.make(args.env).action_space
# action_dim = action_space.high.shape[0]
# act_range = action_space.high
env_before = gym.make(args.env)
env_unwrapped = env_before.unwrapped
# env_unwrapped.observation_space = env_unwrapped.observation_shape
# state_dim = env_unwrapped.observation_space.shape
# action_dim = env_unwrapped.action_space.n
env = Environment(env_before, args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
# action_space = env.action_space
# action_dim = env.get_action_size()
action_dim = env_unwrapped.action_space.shape[0]
act_range = env_unwrapped.action_space.high
print('state: ', state_dim)
print('action: ', action_dim)
print('act range', act_range)
else:
# Standard Environments
# env = Environment(gym.make(args.env), args.consecutive_frames)
# env.reset()
# state_dim = env.get_state_size()
# action_dim = gym.make(args.env).action_space.n
#unreal
env_before = gym.make(args.env)
# env_unwrapped = env_before.unwrapped
# env_unwrapped.observation_space = env_unwrapped.observation_shape
# state_dim = env_unwrapped.observation_space.shape
# action_dim = env_unwrapped.action_space.n
env = Environment(env_before, args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_dim = env.get_action_size()
print('state: ', state_dim)
print('action: ', action_dim)
# state_dim= (640,380)
# action_dim =3
# Pick algorithm to train
print('args type: ', args.type)
if (args.type == "DDQN"):
algo = DDQN(action_dim, state_dim, args)
elif (args.type == "A2C"):
algo = A2C(action_dim, state_dim, args.consecutive_frames)
elif (args.type == "A3C"):
algo = A3C(action_dim,
state_dim,
args.consecutive_frames,
is_atari=args.is_atari)
elif (args.type == "DDPG"):
algo = DDPG(args, action_dim, state_dim, act_range,
args.consecutive_frames)
if args.pretrain:
print('pretrain')
algo.load_weights(args.weights_path)
# Train
stats = algo.train(env, args, summary_writer)
# Export results to CSV
if (args.gather_stats):
df = pd.DataFrame(np.array(stats))
df.to_csv(args.type + "/logs.csv",
header=['Episode', 'Mean', 'Stddev'],
float_format='%10.5f')
# Save weights and close environments
exp_dir = '{}/models/'.format(args.type)
if not os.path.exists(exp_dir):
os.makedirs(exp_dir)
export_path = '{}{}_ENV_{}_NB_EP_{}_BS_{}'.format(exp_dir, args.type,
args.env,
args.nb_episodes,
args.batch_size)
algo.save_weights(export_path)
env.env.close()
def main(args=None):
# Parse arguments
if args is None:
args = sys.argv[1:]
args = parse_args(args)
# Check if a GPU ID was set
if args.gpu:
os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu
set_session(get_session())
summary_writer = tf.summary.FileWriter(args.type + "/tensorboard_" +
args.env)
# Environment Initialization
if (args.is_atari):
# Atari Environment Wrapper
env = AtariEnvironment(args)
state_dim = env.get_state_size()
action_dim = env.get_action_size()
elif (args.type == "DDPG"):
# Continuous Environments Wrapper
env = Environment(gym.make(args.env), args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_space = gym.make(args.env).action_space
action_dim = action_space.high.shape[0]
act_range = action_space.high
else:
# Standard Environments
env = Environment(gym.make(args.env), args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_dim = gym.make(args.env).action_space.n
# Pick algorithm to train
if (args.type == "DDQN"):
algo = DDQN(action_dim, state_dim, args)
elif (args.type == "A2C"):
algo = A2C(action_dim, state_dim, args.consecutive_frames)
elif (args.type == "A3C"):
algo = A3C(action_dim, state_dim, args.consecutive_frames)
elif (args.type == "DDPG"):
algo = DDPG(action_dim, state_dim, act_range, args.consecutive_frames)
# Train
stats = algo.train(env, args, summary_writer)
# Export results to CSV
if (args.gather_stats):
df = pd.DataFrame(np.array(stats))
df.to_csv(args.type + "/logs.csv",
header=['Episode', 'Mean', 'Stddev'],
float_format='%10.5f')
# Display agent
old_state, time = env.reset(), 0
while True:
env.render()
a = algo.policy_action(old_state)
old_state, r, done, _ = env.step(a)
time += 1
if done: env.reset()
import json
import tensorflow as tf
from DDQN.ddqn import DDQN
import os
import time
params = json.load(open("params.json"))
env = recons_env.scanner_env(params)
state_dim = (84, 84, 3)
action_dim = 4
env.reset()
writer = tf.summary.create_file_writer("logs/test")
algo = DDQN(action_dim, state_dim, params["train"])
t0 = time.time()
stats = algo.train(env, params["train"]["nb_episodes"],
params["train"]["batch_size"], writer)
print("It took", time.time() - t0)
exp_dir = 'test/models/'
if not os.path.exists(exp_dir): os.makedirs(exp_dir)
export_path = exp_dir + 'final.h5'
algo.save_weights(export_path)
exp_dir = 'test/stats/'
if not os.path.exists(exp_dir): os.makedirs(exp_dir)
np.savetxt(exp_dir + "test.txt", stats)
def main(args=None):
# Parse arguments 解析参数
if args is None:
args = sys.argv[1:]
args = parse_args(args)
# Check if a GPU ID was set 检查是否设置了GPU ID
if args.gpu:
os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu
set_session(get_session())
summary_writer = tf.summary.FileWriter(args.type + "/tensorboard_" +
args.env)
# Environment Initialization 环境初始化
if (args.is_atari):
# Atari Environment Wrapper Atari环境包装器
env = AtariEnvironment(args)
state_dim = env.get_state_size()
action_dim = env.get_action_size()
elif (args.type == "DDPG"):
# Continuous Environments Wrapper 连续环境包装
env = Environment(gym.make(args.env), args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_space = gym.make(args.env).action_space
action_dim = action_space.high.shape[0]
act_range = action_space.high
else:
# Standard Environments 标准环境
env = Environment(gym.make(args.env), args.consecutive_frames)
env.reset()
state_dim = env.get_state_size()
action_dim = gym.make(args.env).action_space.n
# Pick algorithm to train 选择算法进行训练
if (args.type == "DDQN"):
algo = DDQN(action_dim, state_dim, args)
elif (args.type == "A2C"):
algo = A2C(action_dim, state_dim, args.consecutive_frames)
elif (args.type == "A3C"):
algo = A3C(action_dim,
state_dim,
args.consecutive_frames,
is_atari=args.is_atari)
elif (args.type == "DDPG"):
algo = DDPG(action_dim, state_dim, act_range, args.consecutive_frames)
# Train 训练
stats = algo.train(env, args, summary_writer)
# Export results to CSV 将结果导出到CSV
if (args.gather_stats):
df = pd.DataFrame(np.array(stats))
df.to_csv(args.type + "/logs.csv",
header=['Episode', 'Mean', 'Stddev'],
float_format='%10.5f')
# Save weights and close environments 保存权重并关闭环境
exp_dir = '{}/models/'.format(args.type)
if not os.path.exists(exp_dir):
os.makedirs(exp_dir)
export_path = '{}{}_ENV_{}_NB_EP_{}_BS_{}'.format(exp_dir, args.type,
args.env,
args.nb_episodes,
args.batch_size)
algo.save_weights(export_path)
env.env.close()