def _train_epoch(self, epoch_name: str, handle):
"""A single training epoch, which loops over the number of mini-batches."""
loss = 0
accuracy = 0
format_str = ('%s: loss: %.5f | acc: %.5f | examples/sec: %.2f')
start_time = time.time()
# TODO: refactor: replace this if-else and for-loop block with while-loop and try-except
steps = self.hparams.train_batch_num
for step in range(steps):
fetch_list = [
self.loss_op, self.accuracy_op, self.train_op, self.merged
]
batch_result = self.sess.run(fetch_list,
feed_dict={self.data.handle: handle})
loss += batch_result[0]
accuracy += batch_result[1]
self.writer.add_summary(batch_result[3], self._global_step)
self._global_step += 1
if step % self.hparams.log_step == 0:
log.info('Running %s, batch %d/%d. Loss: %.4f' %
(epoch_name, step, steps, batch_result[0]))
instance_per_sec = steps * self.hparams.batch_size / (time.time() -
start_time)
loss = loss / steps
accuracy = accuracy / steps
log.infov(format_str % ('Training', loss, accuracy, instance_per_sec))
def log_step_message(self, step, accuracy, step_time):
step_time = max(step_time, 0.001)
instance_per_sec = self.batch_size / step_time
log.infov((
f" [test {step:4d}] "
f"Acc.: {accuracy * 100:.2f}% "
f"({step_time:.3f} sec/batch, {instance_per_sec:.3f} instances/sec) "
))
def extract_files(archive, temp_dir):
if os.path.exists(temp_dir):
log.infov("Found existing xyz files at {}, SKIP Extraction!".format(temp_dir))
else:
os.mkdir(temp_dir)
log.info("Extracting files to {} ...".format(temp_dir))
tar = tarfile.open(archive, 'r:*')
tar.extractall(temp_dir)
tar.close()
log.info("Extraction complete.")
def report(self):
log.info("Computing scores...")
correct_prediction_nr = 0
count_nr = 0
correct_prediction_r = 0
count_r = 0
for pred, gt in zip(self._predictions, self._groundtruths):
for i in range(pred.shape[0]):
# relational
if np.argmax(gt[i, :]) < NUM_COLOR:
count_r += 1
if np.argmax(pred[i, :]) == np.argmax(gt[i, :]):
correct_prediction_r += 1
# non-relational
else:
count_nr += 1
if np.argmax(pred[i, :]) == np.argmax(gt[i, :]):
correct_prediction_nr += 1
avg_nr = float(correct_prediction_nr) / count_nr
log.infov("Average accuracy of non-relational questions: {}%".format(
avg_nr * 100))
avg_r = float(correct_prediction_r) / count_r
log.infov("Average accuracy of relational questions: {}%".format(
avg_r * 100))
avg = float(correct_prediction_r + correct_prediction_nr) / (count_r +
count_nr)
log.infov("Average accuracy: {}%".format(avg * 100))
def main():
log.infov('[PSO] Starting PSO algorithm')
pop = toolbox.population(n=POP_SIZE)
stats = tools.Statistics(lambda ind: ind.fitness.values)
stats.register("avg", numpy.mean)
stats.register("std", numpy.std)
stats.register("min", numpy.min)
stats.register("max", numpy.max)
logbook = tools.Logbook()
logbook.header = ["gen", "evals"] + stats.fields
# Maximum generations
MAX_GEN = args.max_gen
best = None
for g in range(MAX_GEN):
fitnesses = toolbox.map(toolbox.evaluate, pop)
for part, fit in zip(pop, fitnesses):
part.fitness.values = (fit['fitness'], )
if not part.best or part.best.fitness < part.fitness:
part.best = creator.Particle(part)
part.best.fitness.values = part.fitness.values
if not best or best.fitness < part.fitness:
best = creator.Particle(part)
best.fitness.values = part.fitness.values
for part in pop:
toolbox.update(part, best)
# Gather all the fitnesses in one list and print the stats
logbook.record(gen=g, evals=len(pop), **stats.compile(pop))
log.info(logbook.stream)
result = {
'gen{}'.format(g): {
'best_params': [best[i] for i in range(len(best))],
'best_fitness': best.fitness.values[0],
}
}
if g == 0:
results = result
else:
results.update(result)
results_IO.to_pickle(results)
logger.log({
'generation': g,
'best_params': [best[i] for i in range(len(best))],
'best_fitness': best.fitness.values[0],
})
logger.write(display=False)
log.infov('best ={}'.format(best.fitness.values[0]))
log.infov('best parm :{}'.format([best[i] for i in range(len(best))]))
return pop, logbook, best
def train(self):
log.infov("Training Starts!")
max_steps = 50000 # total iterations
output_save_step = 5000
for s in range(max_steps):
step, accuracy, summary, loss, data_time, foward_time = self.run_single_step(
) # each training step
# periodic inference
if s % 100 == 0:
self.train_writer.add_summary(summary, global_step=step)
accuracy_val, summary_val = self.run_test() # evaluation
self.val_writer.add_summary(summary_val, global_step=step)
self.log_step_message(step, accuracy, accuracy_val, loss,
data_time, foward_time)
if s % output_save_step == 0:
log.infov("Saved checkpoint at %d", s)
self.saver.save(self.session,
os.path.join(self.train_dir, 'model'),
global_step=step)
def eval_run(self):
# load checkpoint
if self.checkpoint_path:
self.saver.restore(self.session, self.checkpoint_path)
log.info("Loaded from checkpoint!")
log.infov("Start 1-epoch Inference and Evaluation")
log.info("# of examples = %d", len(self.dataset))
length_dataset = len(self.dataset)
max_steps = int(length_dataset / self.batch_size) + 1
log.info("max_steps = %d", max_steps)
coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(self.session,
coord=coord,
start=True)
evaler = EvalManager()
try:
for s in range(max_steps):
step, loss, step_time, prediction_pred, prediction_gt = \
self.run_single_step(self.batch)
self.log_step_message(s, loss, step_time)
evaler.add_batch(prediction_pred, prediction_gt)
except Exception as e:
coord.request_stop(e)
coord.request_stop()
try:
coord.join(threads, stop_grace_period_secs=3)
except RuntimeError as e:
log.warn(str(e))
evaler.report()
log.infov("Evaluation complete.")
def train(self):
with self.graph.as_default():
# TODO: add resume option
# TODO: add exception handler
train_handle = self.sess.run(
self.data.train_iterator.string_handle())
valid_handle = self.sess.run(
self.data.valid_iterator.string_handle())
for epoch in range(self.hparams.epoch_num):
self.sess.run(self.data.train_iterator.initializer)
self.sess.run(self.data.valid_iterator.initializer)
log.infov('Epoch %i' % epoch)
#Train
self._train_epoch('epoch %i (training)' % epoch, train_handle)
#Validate
y_score, y = self._evaluate('epoch %i (evaluating)' % epoch,
valid_handle)
# call roc
y_score = np.concatenate(y_score, axis=0)
y = np.concatenate(y, axis=0)
plot_roc(y_score, y)
def _evaluate(self, epoch_name: str, handle):
"""Evaluation epoch."""
loss = 0
accuracy = 0
f1_scores = []
y_score = []
y = []
format_str = (
'%s: loss: %.5f | acc: %.5f | examples/sec: %.2f \n f1 scores: disorderd %0.5f | 1T %0.5f | 2H %0.5f'
)
steps = self.hparams.valid_batch_num
start_time = time.time()
for step in range(steps):
fetch_list = [
self.loss_op, self.accuracy_op, self.measure, self.y_score,
self.y
]
batch_result = self.sess.run(fetch_list,
feed_dict={self.data.handle: handle})
loss += batch_result[0]
accuracy += batch_result[1]
f1_scores.append(batch_result[2]['f1'])
y_score.append(batch_result[3])
y.append(batch_result[4])
instance_per_sec = steps * self.hparams.batch_size / (time.time() -
start_time)
loss = loss / steps
accuracy = accuracy / steps
f1_scores = np.sum(f1_scores, axis=0) / steps
log.infov(format_str % ('Validation', loss, accuracy, instance_per_sec,
f1_scores[0], f1_scores[1], f1_scores[2]))
return y_score, y
def oscillator_nw(position_vector,
max_time=10.0,
fitness_option=6,
plot=False,
log_dis=False,
render=False,
monitor_path=None,
save_plot_path=None):
if log_dis:
log.infov(
'[OSC]-------------------------------------------------------------'
)
log.infov('[OSC] Run in multiprocessing({})'.format(os.getpid()))
log.infov('[OSC] Running oscillator_2.oscillator_nw')
log.info('[OSC] Printing chromosome')
log.info('[OSC] {0}'.format(position_vector))
log.info('[OSC] Started monitoring thread')
# Start the monitoring thread
env = gym.make('CellrobotEnv-v0')
# For plots - not needed now
if plot:
o1_list = list()
o2_list = list()
o3_list = list()
o4_list = list()
o5_list = list()
o6_list = list()
o7_list = list()
o8_list = list()
o9_list = list()
o10_list = list()
o11_list = list()
o12_list = list()
o13_list = list()
t_list = list()
CPG_controller = CPG_network(position_vector)
# Set monitor thread
monitor_thread = RobotMonitorThread(env, render, monitor_path=monitor_path)
# Set robot API
robot_handle = CRbot(
env,
monitor_thread,
sync_sleep_time=0.01,
interpolation=False,
fraction_max_speed=0.01,
wait=False,
)
# Note the current position
start_pos_x = monitor_thread.x
start_pos_y = monitor_thread.y
start_pos_z = monitor_thread.z
# Start the monitoring thread
monitor_thread.start()
# Set init angles
# robot_handle.set_angles_slow(target_angles=initial_bias_angles, duration=2.0, step=0.01)
# Sleep for 2 seconds to let any oscillations to die down
#time.sleep(2.0)
# Reset the timer of the monitor
monitor_thread.reset_timer()
# New variable for logging up time, since monitor thread is not accurate some times
up_t = 0.0
dt = 0.01
for t in np.arange(0.0, max_time, dt):
# Increment the up time variable
up_t = t
output_list = CPG_controller.output(state=None)
# Set the joint positions
current_angles = {
'cell0': output_list[1],
'cell1': output_list[2],
'cell2': output_list[3],
'cell3': output_list[4],
'cell4': output_list[5],
'cell5': output_list[6],
'cell6': output_list[7],
'cell7': output_list[8],
'cell8': output_list[9],
'cell9': output_list[10],
'cell10': output_list[11],
'cell11': output_list[12],
'cell12': output_list[13]
}
robot_handle.set_angles(current_angles)
time.sleep(dt)
# Check if the robot has fallen
if monitor_thread.fallen:
break
# For plots - not needed now
if plot:
o1_list.append(output_list[1])
o2_list.append(output_list[2])
o3_list.append(output_list[3])
o4_list.append(output_list[4])
o5_list.append(output_list[5])
o6_list.append(output_list[6])
o7_list.append(output_list[7])
o8_list.append(output_list[8])
o9_list.append(output_list[9])
o10_list.append(output_list[10])
o11_list.append(output_list[11])
o12_list.append(output_list[12])
o13_list.append(output_list[13])
t_list.append(t)
if log_dis:
log.info('[OSC] Accurate up time: {0}'.format(up_t))
# Outside the loop, it means that either the robot has fallen or the max_time has elapsed
# Find out the end position of the robot
end_pos_x = monitor_thread.x
end_pos_y = monitor_thread.y
end_pos_z = monitor_thread.z
# Find the average height
avg_z = monitor_thread.avg_z
# Find the up time
# up_time = monitor_thread.up_time
up_time = up_t
# Calculate the fitness
if up_time == 0.0:
fitness = 0.0
if log_dis:
log('[OSC] up_t==0 so fitness is set to 0.0')
else:
fitness = calc_fitness(start_x=start_pos_x,
start_y=start_pos_y,
start_z=start_pos_z,
end_x=end_pos_x,
end_y=end_pos_y,
end_z=end_pos_z,
avg_z=avg_z,
up_time=up_time,
fitness_option=fitness_option)
if log_dis:
if not monitor_thread.fallen:
log.info("[OSC] Robot has not fallen")
else:
log.info("[OSC] Robot has fallen")
log.info('[OSC] Calculated fitness: {0}'.format(fitness))
# Different from original script
# Fetch the values of the evaluation metrics
fallen = monitor_thread.fallen
up = up_time # Using a more accurate up time than monitor_thread.up_time,
x_distance = end_pos_x - start_pos_x
abs_y_deviation = end_pos_y
avg_footstep_x = None
var_torso_alpha = monitor_thread.obs[3]
var_torso_beta = monitor_thread.obs[4]
var_torso_gamma = monitor_thread.obs[5]
# Stop the monitoring thread
monitor_thread.stop()
# Close the VREP connection
robot_handle.cleanup()
# For plots - not needed now
if plot:
ax1 = plt.subplot(611)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o2_list, color='green', ls='--', label='o_2')
plt.plot(t_list, o3_list, color='green', label='o_3')
plt.grid()
plt.legend()
ax2 = plt.subplot(612, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o4_list, color='blue', ls='--', label='o_4')
plt.plot(t_list, o5_list, color='blue', label='o_5')
plt.grid()
plt.legend()
ax3 = plt.subplot(613, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o6_list, color='black', ls='--', label='o_6')
plt.plot(t_list, o7_list, color='black', label='o_7')
plt.grid()
plt.legend()
ax4 = plt.subplot(614, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o8_list, color='cyan', ls='--', label='o_8')
plt.plot(t_list, o9_list, color='cyan', label='o_9')
plt.grid()
plt.legend()
ax5 = plt.subplot(615, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o10_list, color='orange', ls='--', label='o_10')
plt.plot(t_list, o11_list, color='orange', label='o_11')
plt.grid()
plt.legend()
ax6 = plt.subplot(616, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o12_list, color='brown', ls='--', label='o_12')
plt.plot(t_list, o13_list, color='brown', label='o_13')
plt.grid()
plt.legend()
if save_plot_path is not None:
plt.savefig(save_plot_path)
else:
plt.show()
# Different from original script
# Return the evaluation metrics
return {
'fitness': fitness,
'fallen': fallen,
'up': up,
'x_distance': x_distance,
'abs_y_deviation': abs_y_deviation,
'avg_footstep_x': avg_footstep_x,
'var_torso_alpha': var_torso_alpha,
'var_torso_beta': var_torso_beta,
'var_torso_gamma': var_torso_gamma
}
toolbox.register("select", tools.selTournament, tournsize=3)
# Size of the population
POP_SIZE = 200
# Maximum generations
MAX_GEN = 31
# Create an initial population of `POP_SIZE` individuals (where each individual is a list of floats)
pop = toolbox.population(n=POP_SIZE)
# CXPB is the probability with which two individuals are crossed
# MUTPB is the probability for mutating an individual
CXPB, MUTPB = 0.8, 0.1
log.infov('[GA] Starting genetic algorithm')
# Evaluate the entire population and store the fitness of each individual
log.infov('[GA] Finding the fitness of individuals in the initial generation')
fitnesses = list(map(toolbox.evaluate, pop))
for ind, fit in zip(pop, fitnesses):
print(ind, fit)
ind.fitness.values = (fit, )
# Extracting all the fitnesses
fits = [ind.fitness.values[0] for ind in pop]
# Variable keeping track of the number of generations
g = 0
best_ind_ever = None
def main():
random.seed(64)
# Create an initial population of `POP_SIZE` individuals (where each individual is a list of floats)
pop = toolbox.population(n=POP_SIZE)
# CXPB is the probability with which two individuals are crossed
# MUTPB is the probability for mutating an individual
CXPB, MUTPB = args.CXPB, args.MUTPB
log.infov('[GA] Starting genetic algorithm')
# Evaluate the entire population and store the fitness of each individual
log.infov(
'[GA] Finding the fitness of individuals in the initial generation')
fitnesses = list(toolbox.map(toolbox.evaluate, pop))
#log_save_info(fitnesses)
for ind, fit in zip(pop, fitnesses):
#print(ind, fit)
ind.fitness.values = (fit['fitness'], )
# Extracting all the fitnesses
fits = [ind.fitness.values[0] for ind in pop]
# Variable keeping track of the number of generations
g = 0
best_ind_ever = None
best_fitness_ever = 0.0
# Begin the evolution
while max(fits) < 100 and g < MAX_GEN:
# A new generation
g = g + 1
log.infov('[GA{}] Running generation {}'.format(g, g))
# Select the next generation individuals
log.info('[GA{}] Selecting the next generation'.format(g))
offspring = toolbox.select(pop, len(pop))
# Clone the selected individuals
offspring = list(toolbox.map(toolbox.clone, offspring))
# Apply crossover and mutation on the offspring
for child1, child2 in zip(offspring[::2], offspring[1::2]):
# cross two individuals with probability CXPB
if random.random() < CXPB:
toolbox.mate(child1, child2)
# fitness values of the children
# must be recalculated later
del child1.fitness.values
del child2.fitness.values
for mutant in offspring:
# mutate an individual with probability MUTPB
if random.random() < MUTPB:
toolbox.mutate(mutant)
del mutant.fitness.values
# Since the content of some of our offspring changed during the last step, we now need to
# re-evaluate their fitnesses. To save time and resources, we just map those offspring which
# fitnesses were marked invalid.
# Evaluate the individuals with an invalid fitness
invalid_ind = [ind for ind in offspring if not ind.fitness.valid]
fitnesses = toolbox.map(toolbox.evaluate, invalid_ind)
for ind, fit in zip(invalid_ind, fitnesses):
ind.fitness.values = (fit['fitness'], )
log.info('[GA{}] Evaluated {} individuals (invalid fitness)'.format(
g, len(invalid_ind)))
# The population is entirely replaced by the offspring
pop[:] = offspring
# Gather all the fitnesses in one list and print the stats
fits = [ind.fitness.values[0] for ind in pop]
length = len(pop)
mean = sum(fits) / length
sum2 = sum(x * x for x in fits)
std = abs(sum2 / length - mean**2)**0.5
log.info('[GA{}] Results for generation {}'.format(g, g))
log.info('[GA{}] Min {}'.format(g, min(fits)))
log.info('[GA{}] Max {}'.format(g, max(fits)))
log.info('[GA{}] Avg {}'.format(g, mean))
log.info('[GA{}] Std {}'.format(g, std))
best_ind_g = tools.selBest(pop, 1)[0]
# Store the best individual over all generations
if best_ind_g.fitness.values[0] > best_fitness_ever:
best_fitness_ever = best_ind_g.fitness.values[0]
best_ind_ever = best_ind_g
log.info('[GA{}] Best individual for generation {}: {}, {}'.format(
g, g, best_ind_g, best_ind_g.fitness.values[0]))
log.info('[GA{}] Best individual ever till now: {}, {}'.format(
g, best_ind_ever, best_fitness_ever))
best_pram_list = [best_ind_g[_] for _ in range(len(best_ind_g))]
best_pram_ever_list = [
best_ind_ever[_] for _ in range(len(best_ind_ever))
]
result = {
'gen{}'.format(g): {
'best_params': best_pram_list,
'best_fitness': best_ind_g.fitness.values[0],
'best_iparams_ever': best_pram_ever_list,
'best_fitness_ever': best_fitness_ever,
}
}
if g == 1:
results = result
else:
results.update(result)
results_IO.to_pickle(results)
logger.log({
'generation': g,
'best_params': best_ind_g,
'best_fitness': best_ind_g.fitness.values[0],
'best_iparams_ever': best_ind_ever,
'best_fitness_ever': best_fitness_ever,
})
logger.write(display=False)
log.infov(
'[GA{0}] ############################# End of generation #############################'
.format(g))
log.infov('===================== End of evolution =====================')
best_ind = tools.selBest(pop, 1)[0]
log.infov('Best individual in the population: %s, %s' %
(best_ind, best_ind.fitness.values[0]))
log.infov('Best individual ever: %s, %s' %
(best_ind_ever, best_fitness_ever))
CPG_parm_num = 9
oscillator_nw = partial(oscillator_nw, env_name=env_name)
# Set the logging variables
# This also creates a new log file
# Create log files
log_dir = configure_log_dir(env_name, txt=log_name, No_time=False)
logging_output(log_dir)
logger = LoggerCsv(log_dir, csvname='log_results')
results_IO = IO(os.path.join(log_dir, 'results.pkl'))
args_IO = IO(os.path.join(log_dir, 'args.pkl')).to_pickle(args)
# Create the position bounds of the individual
log.info('[System] parmeters: {}'.format(args))
log.infov('[GA] Running ga_2')
log.infov('[GA] Creating position bounds')
FLT_MIN_KF, FLT_MAX_KF = 0.2, 1.0
FLT_MIN_GAIN0, FLT_MAX_GAIN0 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN1, FLT_MAX_GAIN1 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN2, FLT_MAX_GAIN2 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN3, FLT_MAX_GAIN3 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN4, FLT_MAX_GAIN4 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN5, FLT_MAX_GAIN5 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN6, FLT_MAX_GAIN6 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN7, FLT_MAX_GAIN7 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_GAIN8, FLT_MAX_GAIN8 = -1 * args.gain_max, 1 * args.gain_max
FLT_MIN_BIAS0, FLT_MAX_BIAS0 = -math.radians(90), math.radians(90)
FLT_MIN_BIAS1, FLT_MAX_BIAS1 = -math.radians(90), math.radians(90)
FLT_MIN_BIAS2, FLT_MAX_BIAS2 = -math.radians(90), math.radians(90)
def oscillator_nw(position_vector, max_time=10.0, fitness_option=6, plot = False, log_dis = False, render = False, monitor_path=None, save_plot_path = None):
if log_dis:
log.infov('[OSC]-------------------------------------------------------------')
log.infov('[OSC] Run in multiprocessing({})'.format(os.getpid()))
log.infov('[OSC] Running oscillator_2.oscillator_nw')
log.info('[OSC] Printing chromosome')
log.info('[OSC] {0}'.format(position_vector))
log.info('[OSC] Started monitoring thread')
# Start the monitoring thread
env = gym.make('CellrobotEnv-v0')
# Extract the elements from the position vector
kf = position_vector[0]
GAIN0 = position_vector[1]
GAIN1 = position_vector[2]
GAIN2 = position_vector[3]
GAIN3 = position_vector[4]
GAIN4 = position_vector[5]
GAIN5 = position_vector[6]
GAIN6 = position_vector[7]
GAIN7 = position_vector[8]
GAIN8 = position_vector[9]
GAIN9 = position_vector[10]
GAIN10 = position_vector[11]
GAIN11 = position_vector[12]
GAIN12 = position_vector[13]
BIAS0 = position_vector[14]
BIAS1 = position_vector[15]
BIAS2 = position_vector[16]
BIAS3 = position_vector[17]
BIAS4 = position_vector[18]
BIAS5 = position_vector[19]
BIAS6 = position_vector[20]
BIAS7 = position_vector[21]
BIAS8 = position_vector[22]
BIAS9 = position_vector[23]
BIAS10 = position_vector[24]
BIAS11 = position_vector[25]
BIAS12 = position_vector[26]
osillator = matsuoka_oscillator(kf)
osillator_fun = osillator.oscillator_fun
# Variables
# Oscillator 1 (pacemaker)
u1_1, u2_1, v1_1, v2_1, y1_1, y2_1, o_1, gain_1, bias_1 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0, 0.0
# Oscillator 2 --cell0
u1_2, u2_2, v1_2, v2_2, y1_2, y2_2, o_2, gain_2, bias_2 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN0, BIAS0
# Oscillator 3 --cell1
u1_3, u2_3, v1_3, v2_3, y1_3, y2_3, o_3, gain_3, bias_3 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN1, BIAS1
# Oscillator 4 --cell2
u1_4, u2_4, v1_4, v2_4, y1_4, y2_4, o_4, gain_4, bias_4 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN2, BIAS2
# Oscillator 5 --cell3
u1_5, u2_5, v1_5, v2_5, y1_5, y2_5, o_5, gain_5, bias_5 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN3, BIAS3
# Oscillator 6 --cell4
u1_6, u2_6, v1_6, v2_6, y1_6, y2_6, o_6, gain_6, bias_6 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN4, BIAS4
# Oscillator 7 --cell5
u1_7, u2_7, v1_7, v2_7, y1_7, y2_7, o_7, gain_7, bias_7 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN5, BIAS5
# Oscillator 8 --cell6
u1_8, u2_8, v1_8, v2_8, y1_8, y2_8, o_8, gain_8, bias_8 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN6, BIAS6
# Oscillator 9 --cell7
u1_9, u2_9, v1_9, v2_9, y1_9, y2_9, o_9, gain_9, bias_9 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN7, BIAS7
# Oscillator 10 --cell8
u1_10, u2_10, v1_10, v2_10, y1_10, y2_10, o_10, gain_10, bias_10 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN8, BIAS8
# Oscillator 11 --cell9
u1_11, u2_11, v1_11, v2_11, y1_11, y2_11, o_11, gain_11, bias_11 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN9, BIAS9
# Oscillator 12 --cell10
u1_12, u2_12, v1_12, v2_12, y1_12, y2_12, o_12, gain_12, bias_12 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN10, BIAS10
# Oscillator 13 --cell11
u1_13, u2_13, v1_13, v2_13, y1_13, y2_13, o_13, gain_13, bias_13 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN11, BIAS11
# Oscillator 14 --cell12
u1_14, u2_14, v1_14, v2_14, y1_14, y2_14, o_14, gain_14, bias_14 = 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, GAIN12, BIAS12
# For plots - not needed now
if plot:
o1_list = list()
o2_list = list()
o3_list = list()
o4_list = list()
o5_list = list()
o6_list = list()
o7_list = list()
o8_list = list()
o9_list = list()
o10_list = list()
o11_list = list()
o12_list = list()
o13_list = list()
t_list = list()
# Set the joints to the initial bias positions - use slow angle setter
initial_bias_angles = {'cell0':BIAS0, 'cell1':BIAS1,'cell2':BIAS2,'cell3':BIAS3,'cell4':BIAS4,
'cell5':BIAS5, 'cell6':BIAS6,'cell7':BIAS7,'cell8':BIAS8,'cell9':BIAS9,
'cell10':BIAS10, 'cell11':BIAS11,'cell12':BIAS12
}
# Set monitor thread
monitor_thread = RobotMonitorThread(env, render, monitor_path=monitor_path)
# Set robot API
robot_handle = CRbot(env, monitor_thread, sync_sleep_time=0.01, interpolation=False, fraction_max_speed=0.01,
wait=False, )
# Note the current position
start_pos_x = monitor_thread.x
start_pos_y = monitor_thread.y
start_pos_z = monitor_thread.z
# Start the monitoring thread
monitor_thread.start()
# Set init angles
# robot_handle.set_angles_slow(target_angles=initial_bias_angles, duration=2.0, step=0.01)
# Sleep for 2 seconds to let any oscillations to die down
#time.sleep(2.0)
# Reset the timer of the monitor
monitor_thread.reset_timer()
# New variable for logging up time, since monitor thread is not accurate some times
up_t = 0.0
dt =0.01
for t in np.arange(0.0, max_time, dt):
# Increment the up time variable
up_t = t
#-----------------------------------------------------------------------------------------------------
# ---------------------------------------NETWORK START-----------------------------------------------
# -----------------------------------------------------------------------------------------------------
# Calculate next state of oscillator 1 (pacemaker)
f1_1, f2_1 = 0.0, 0.0
s1_1, s2_1 = 0.0, 0.0
u1_1, u2_1, v1_1, v2_1, y1_1, y2_1, o_1 = osillator_fun(u1=u1_1, u2=u2_1, v1=v1_1, v2=v2_1, y1=y1_1, y2=y2_1,
f1=f1_1, f2=f2_1, s1=s1_1, s2=s2_1,
bias=bias_1, gain=gain_1, )
# center
# Calculate next state of oscillator 2 --cell0
# w_ij -> j=1 (oscillator 1) is master, i=2 (oscillator 2) is slave
w_21 = 1.0
f1_2, f2_2 = 0.0, 0.0
s1_2, s2_2 = w_21 * u1_1, w_21 * u2_1
u1_2, u2_2, v1_2, v2_2, y1_2, y2_2, o_2 = osillator_fun(u1=u1_2, u2=u2_2, v1=v1_2, v2=v2_2, y1=y1_2, y2=y2_2,
f1=f1_2, f2=f2_2, s1=s1_2, s2=s2_2,
bias=bias_2, gain=gain_2)
# Left forward 1
# Calculate next state of oscillator 3 --cell1
# w_ij -> j=1 (oscillator 2) is master, i=3 (oscillator 3) is slave
w_32 = 1.0
f1_3, f2_3 = 0.0, 0.0
s1_3, s2_3 = w_32 * u1_1, w_32 * u2_1
u1_3, u2_3, v1_3, v2_3, y1_3, y2_3, o_3 = osillator_fun(u1=u1_3, u2=u2_3, v1=v1_3, v2=v2_3, y1=y1_3, y2=y2_3,
f1=f1_3, f2=f2_3, s1=s1_3, s2=s2_3,
bias=bias_3, gain=gain_3)
# Left back 1
# Calculate next state of oscillator 4 --cell2
# w_ij -> j=2 (oscillator 2) is master, i=4 (oscillator 4) is slave
w_42 = 1.0
f1_4, f2_4 = 0.0, 0.0
s1_4, s2_4 = w_42 * u1_2, w_42 * u2_2 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_4, u2_4, v1_4, v2_4, y1_4, y2_4, o_4 = osillator_fun(u1=u1_4, u2=u2_4, v1=v1_4, v2=v2_4, y1=y1_4, y2=y2_4,
f1=f1_4, f2=f2_4, s1=s1_4, s2=s2_4,
bias=bias_4, gain=gain_4)
# Right forward 1
# Calculate next state of oscillator 5 --cell3
# w_ij -> j=3 (oscillator 3) is master, i=5 (oscillator 5) is slave
w_52 = -1.0
f1_5, f2_5 = 0.0, 0.0
s1_5, s2_5 = w_52 * u1_3, w_52 * u2_3 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_5, u2_5, v1_5, v2_5, y1_5, y2_5, o_5 = osillator_fun(u1=u1_5, u2=u2_5, v1=v1_5, v2=v2_5, y1=y1_5, y2=y2_5,
f1=f1_5, f2=f2_5, s1=s1_5, s2=s2_5,
bias=bias_5, gain=gain_5)
# Right back1
# Calculate next state of oscillator 6 --cell4
# w_ij -> j=2 (oscillator 2) is master, i=6 (oscillator 6) is slave
w_62 = -1.0
f1_6, f2_6 = 0.0, 0.0
s1_6, s2_6 = w_62 * u1_2, w_62 * u2_2 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_6, u2_6, v1_6, v2_6, y1_6, y2_6, o_6 = osillator_fun(u1=u1_6, u2=u2_6, v1=v1_6, v2=v2_6, y1=y1_6, y2=y2_6,
f1=f1_6, f2=f2_6, s1=s1_6, s2=s2_6,
bias=bias_6, gain=gain_6)
# Left forward 2
# Calculate next state of oscillator 7 --cell5
# w_ij -> j=3 (oscillator 3) is master, i=7 (oscillator 7) is slave
w_73 = -1.0
f1_7, f2_7 = 0.0, 0.0
s1_7, s2_7 = w_73 * u1_3, w_73 * u2_3 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_7, u2_7, v1_7, v2_7, y1_7, y2_7, o_7 = osillator_fun(u1=u1_7, u2=u2_7, v1=v1_7, v2=v2_7, y1=y1_7, y2=y2_7,
f1=f1_7, f2=f2_7, s1=s1_7, s2=s2_7,
bias=bias_7, gain=gain_7)
# Left foward 3
# Calculate next state of oscillator 8 --cell6
# w_ij -> j=1 (oscillator 3) is master, i=8 (oscillator 8) is slave
w_83 = -1.0
f1_8, f2_8 = 0.0, 0.0
s1_8, s2_8 = w_83 * u1_1, w_83 * u2_1 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_8, u2_8, v1_8, v2_8, y1_8, y2_8, o_8 = osillator_fun(u1=u1_8, u2=u2_8, v1=v1_8, v2=v2_8, y1=y1_8, y2=y2_8,
f1=f1_8, f2=f2_8, s1=s1_8, s2=s2_8,
bias=bias_8, gain=gain_8)
# Left back 2
# Calculate next state of oscillator 9 --cell7
# w_ij -> j=8 (oscillator 4) is master, i=9 (oscillator 9) is slave
w_94 = -1.0
f1_9, f2_9 = 0.0, 0.0
s1_9, s2_9 = w_94 * u1_8, w_94 * u2_8 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_9, u2_9, v1_9, v2_9, y1_9, y2_9, o_9 = osillator_fun(u1=u1_9, u2=u2_9, v1=v1_9, v2=v2_9, y1=y1_9, y2=y2_9,
f1=f1_9, f2=f2_9, s1=s1_9, s2=s2_9,
bias=bias_9, gain=gain_9)
# Left back 3
# Calculate next state of oscillator 10 --cell8
# w_ij -> j=1 (oscillator 4) is master, i=10 (oscillator 10) is slave
w_104 = -1.0
f1_10, f2_10 = 0.0, 0.0
s1_10, s2_10 = w_104 * u1_1, w_104 * u2_1 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_10, u2_10, v1_10, v2_10, y1_10, y2_10, o_10 = osillator_fun(u1=u1_10, u2=u2_10, v1=v1_10, v2=v2_10, y1=y1_10,
y2=y2_10,
f1=f1_10, f2=f2_10, s1=s1_10, s2=s2_10,
bias=bias_10, gain=gain_10)
# Right forward 2
# Calculate next state of oscillator 11 --cell9
# w_ij -> j=10 (oscillator 5) is master, i=11 (oscillator 11) is slave
w_115 = -1.0
f1_11, f2_11 = 0.0, 0.0
s1_11, s2_11 = w_115 * u1_10, w_115 * u2_10 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_11, u2_11, v1_11, v2_11, y1_11, y2_11, o_11 = osillator_fun(u1=u1_11, u2=u2_11, v1=v1_11, v2=v2_11, y1=y1_11,
y2=y2_11,
f1=f1_11, f2=f2_11, s1=s1_11, s2=s2_11,
bias=bias_11, gain=gain_11)
# Right forward 3
# Calculate next state of oscillator 12 --cell10
# w_ij -> j=1 (oscillator 5) is master, i=12 (oscillator 12) is slave
w_125 = -1.0
f1_12, f2_12 = 0.0, 0.0
s1_12, s2_12 = w_125 * u1_1, w_125 * u2_1 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_12, u2_12, v1_12, v2_12, y1_12, y2_12, o_12 = osillator_fun(u1=u1_12, u2=u2_12, v1=v1_12, v2=v2_12, y1=y1_12,
y2=y2_12,
f1=f1_12, f2=f2_12, s1=s1_12, s2=s2_12,
bias=bias_12, gain=gain_12)
# Right back 2
# Calculate next state of oscillator 13 --cell11
# w_ij -> j=1 (oscillator 6) is master, i=13 (oscillator 13) is slave
w_136 = -1.0
f1_13, f2_13 = 0.0, 0.0
s1_13, s2_13 = w_136 * u1_1, w_136 * u2_1 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_13, u2_13, v1_13, v2_13, y1_13, y2_13, o_13 = osillator_fun(u1=u1_13, u2=u2_13, v1=v1_13, v2=v2_13, y1=y1_13,
y2=y2_13,
f1=f1_13, f2=f2_13, s1=s1_13, s2=s2_13,
bias=bias_13, gain=gain_13)
# Right back 3
# Calculate next state of oscillator 14 --cell11
# w_ij -> j=1 (oscillator 6) is master, i=14 (oscillator 14) is slave
w_146 = -1.0
f1_14, f2_14 = 0.0, 0.0
s1_14, s2_14 = w_146 * u1_1, w_146 * u2_1 # s1_i = w_ij*u1_j, s2_i = w_ij*u2_j
u1_14, u2_14, v1_14, v2_14, y1_14, y2_14, o_14 = osillator_fun(u1=u1_14, u2=u2_14, v1=v1_14, v2=v2_14, y1=y1_14,
y2=y2_14,
f1=f1_14, f2=f2_14, s1=s1_14, s2=s2_14,
bias=bias_14, gain=gain_14)
# -----------------------------------------------------------------------------------------------------
# ---------------------------------------NETWORK END--------------------------------------------------
# -----------------------------------------------------------------------------------------------------
# Set the joint positions
current_angles = {'cell0':o_2, 'cell1':o_3,'cell2':o_4,'cell3':o_5,'cell4':o_6,
'cell5':o_7, 'cell6':o_8,'cell7':o_9,'cell8':o_10,'cell9':o_11,
'cell10':o_12, 'cell11':o_13,'cell12':o_14
}
robot_handle.set_angles(current_angles)
time.sleep(dt)
# Check if the robot has fallen
if monitor_thread.fallen:
break
# For plots - not needed now
if plot:
o1_list.append(o_1)
o2_list.append(o_2)
o3_list.append(o_3)
o4_list.append(o_4)
o5_list.append(o_5)
o6_list.append(o_6)
o7_list.append(o_7)
o8_list.append(o_8)
o9_list.append(o_9)
o10_list.append(o_10)
o11_list.append(o_11)
o12_list.append(o_12)
o13_list.append(o_13)
t_list.append(t)
if log_dis:
log.info('[OSC] Accurate up time: {0}'.format(up_t))
# Outside the loop, it means that either the robot has fallen or the max_time has elapsed
# Find out the end position of the robot
end_pos_x = monitor_thread.x
end_pos_y = monitor_thread.y
end_pos_z = monitor_thread.z
# Find the average height
avg_z = monitor_thread.avg_z
# Find the up time
# up_time = monitor_thread.up_time
up_time = up_t
# Calculate the fitness
if up_time == 0.0:
fitness = 0.0
if log_dis:
log('[OSC] up_t==0 so fitness is set to 0.0')
else:
fitness = calc_fitness(start_x=start_pos_x, start_y=start_pos_y, start_z=start_pos_z,
end_x=end_pos_x, end_y=end_pos_y, end_z=end_pos_z,
avg_z=avg_z,
up_time=up_time,
fitness_option=fitness_option
)
if log_dis:
if not monitor_thread.fallen:
log.info("[OSC] Robot has not fallen")
else:
log.info("[OSC] Robot has fallen")
log.info('[OSC] Calculated fitness: {0}'.format(fitness))
# Different from original script
# Fetch the values of the evaluation metrics
fallen = monitor_thread.fallen
up = up_time # Using a more accurate up time than monitor_thread.up_time,
x_distance = end_pos_x - start_pos_x
abs_y_deviation = end_pos_y
avg_footstep_x = None
var_torso_alpha = monitor_thread.obs[3]
var_torso_beta = monitor_thread.obs[4]
var_torso_gamma = monitor_thread.obs[5]
# Stop the monitoring thread
monitor_thread.stop()
# Close the VREP connection
robot_handle.cleanup()
# For plots - not needed now
if plot:
ax1 = plt.subplot(611)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o2_list, color='green', ls='--', label='o_2')
plt.plot(t_list, o3_list, color='green', label='o_3')
plt.grid()
plt.legend()
ax2 = plt.subplot(612, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o4_list, color='blue', ls='--', label='o_4')
plt.plot(t_list, o5_list, color='blue', label='o_5')
plt.grid()
plt.legend()
ax3 = plt.subplot(613, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o6_list, color='black', ls='--', label='o_6')
plt.plot(t_list, o7_list, color='black', label='o_7')
plt.grid()
plt.legend()
ax4 = plt.subplot(614, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o8_list, color='cyan', ls='--', label='o_8')
plt.plot(t_list, o9_list, color='cyan', label='o_9')
plt.grid()
plt.legend()
ax5 = plt.subplot(615, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o10_list, color='orange', ls='--', label='o_10')
plt.plot(t_list, o11_list, color='orange', label='o_11')
plt.grid()
plt.legend()
ax6 = plt.subplot(616, sharex=ax1, sharey=ax1)
plt.plot(t_list, o1_list, color='red', label='o_1')
plt.plot(t_list, o12_list, color='brown', ls='--', label='o_12')
plt.plot(t_list, o13_list, color='brown', label='o_13')
plt.grid()
plt.legend()
if save_plot_path is not None:
plt.savefig(save_plot_path)
else:
plt.show()
# Different from original script
# Return the evaluation metrics
return {'fitness': fitness,
'fallen': fallen,
'up': up,
'x_distance': x_distance,
'abs_y_deviation': abs_y_deviation,
'avg_footstep_x': avg_footstep_x,
'var_torso_alpha': var_torso_alpha,
'var_torso_beta': var_torso_beta,
'var_torso_gamma': var_torso_gamma}
def __init__(self, cfg, dataset):
self.config = cfg
self.train_dir = cfg.train_dir
log.info("self.train_dir = %s", self.train_dir)
# --- input ops ---
self.batch_size = cfg.batch_size
self.dataset = dataset
self.batch = create_input_ops(dataset, self.batch_size, shuffle=False)
# --- create model ---
if cfg.model == 'baseline':
from models.baseline import Model
elif cfg.model == 'rn':
from models.rn import Model
elif cfg.model == 'film':
from models.film import Model
else:
raise ValueError(cfg.model)
log.infov("Using Model class : %s", Model)
self.model = Model(Q_DIM, NUM_ANS, is_train=False)
self.img = tf.placeholder(
name='img',
dtype=tf.float32,
shape=[self.batch_size, cfg.image_size, cfg.image_size, 3],
)
self.q = tf.placeholder(name='q',
dtype=tf.float32,
shape=[cfg.batch_size, Q_DIM])
self.a = tf.placeholder(name='a',
dtype=tf.float32,
shape=[cfg.batch_size, NUM_ANS])
logits = self.model.build(self.img, self.q)
correct_prediction = tf.equal(tf.argmax(logits, 1),
tf.argmax(self.a, 1))
self.accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
self.all_preds = tf.nn.softmax(logits)
self.global_step = tf.contrib.framework.get_or_create_global_step(
graph=None)
self.step_op = tf.no_op(name='step_no_op')
tf.set_random_seed(1234)
session_config = tf.ConfigProto(gpu_options=tf.GPUOptions(
allow_growth=True))
self.session = tf.Session(config=session_config)
# --- checkpoint and monitoring ---
self.saver = tf.train.Saver()
self.checkpoint_path = cfg.checkpoint_path
if self.checkpoint_path is None and self.train_dir:
self.checkpoint_path = tf.train.latest_checkpoint(self.train_dir)
if self.checkpoint_path is None:
log.warn("No checkpoint is given. Just random initialization :-)")
self.session.run(tf.global_variables_initializer())
else:
log.info("Checkpoint path : %s", self.checkpoint_path)
def download_qm9(url, file):
if os.path.exists(file):
log.infov("Found existing QM9 dataset at {}, SKIP downloading!".format(file))
return
wget.download(url, out=file)
def __init__(self, cfg, train_dataset, val_dataset):
self.config = cfg
dataset_base = os.path.basename(os.path.normpath(cfg.dataset_path))
hyper_parameter_str = dataset_base + '_lr_' + str(cfg.learning_rate)
self.train_dir = './train_dir/%s-%s/%s/%s' % (
cfg.model, cfg.prefix, hyper_parameter_str,
time.strftime("%Y%m%d-%H%M%S"))
if not os.path.exists(self.train_dir):
os.makedirs(self.train_dir)
log.infov("Train Dir: %s", self.train_dir)
# --- input ops ---
self.batch_size = cfg.batch_size
self.train_batch = create_input_ops(train_dataset, self.batch_size)
self.val_batch = create_input_ops(val_dataset, self.batch_size)
# --- create model ---
if cfg.model == 'baseline':
from models.baseline import Model
elif cfg.model == 'rn':
from models.rn import Model
elif cfg.model == 'film':
from models.film import Model
else:
raise ValueError(cfg.model)
log.infov("Using Model class : %s", Model)
self.model = Model(Q_DIM, NUM_ANS)
# define placeholders: (image, question, answer)
self.img = tf.placeholder(
name='img',
dtype=tf.float32,
shape=[self.batch_size, cfg.image_size, cfg.image_size, 3],
)
self.q = tf.placeholder(name='q',
dtype=tf.float32,
shape=[cfg.batch_size, Q_DIM])
self.a = tf.placeholder(name='a',
dtype=tf.float32,
shape=[cfg.batch_size, NUM_ANS])
# compute logits and cross-entropy loss
logits = self.model.build(self.img, self.q)
loss = tf.nn.softmax_cross_entropy_with_logits(logits=logits,
labels=self.a)
loss = tf.reduce_mean(loss)
correct_prediction = tf.equal(tf.argmax(logits, 1),
tf.argmax(self.a, 1))
accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
self.all_preds = tf.nn.softmax(logits)
self.loss, self.accuracy = loss, accuracy
tf.summary.scalar("accuracy", self.accuracy)
tf.summary.scalar("loss", self.loss)
# --- optimizer ---
self.global_step = tf.contrib.framework.get_or_create_global_step(
graph=None)
self.learning_rate = cfg.learning_rate
if cfg.lr_weight_decay:
# learning rate scheduling (optional)
self.learning_rate = tf.train.exponential_decay(
self.learning_rate,
global_step=self.global_step,
decay_steps=10000,
decay_rate=0.95,
name='decay_lr')
self.check_op = tf.no_op()
# Adam optimizer
self.optimizer = tf.contrib.layers.optimize_loss(
loss=self.loss,
global_step=self.global_step,
learning_rate=self.learning_rate,
optimizer=tf.train.AdamOptimizer,
clip_gradients=20.0,
name='optimizer_loss')
self.summary_op = tf.summary.merge_all()
self.saver = tf.train.Saver()
self.train_writer = tf.summary.FileWriter(self.train_dir + '/train')
self.val_writer = tf.summary.FileWriter(self.train_dir + '/val')
self.supervisor = tf.train.Supervisor(
logdir=self.train_dir,
is_chief=True,
saver=None,
summary_op=None,
summary_writer=self.train_writer,
save_summaries_secs=100,
global_step=self.global_step,
)
session_config = tf.ConfigProto(gpu_options=tf.GPUOptions(
allow_growth=True))
self.session = self.supervisor.prepare_or_wait_for_session(
config=session_config)
self.ckpt_path = cfg.checkpoint
if self.ckpt_path is not None:
log.info("Checkpoint path: %s", self.ckpt_path)
self.saver.restore(self.session, self.ckpt_path)
log.info(
"Loaded the pretrain parameters from the provided checkpoint path"
)