def dl_progress(count, block_size, total_size):
if ProgressTracker.progbar is None:
if total_size is -1:
total_size = None
ProgressTracker.progbar = Progbar(total_size)
else:
ProgressTracker.progbar.update(count * block_size)
def __init__(self, env, config):
Dueling.__init__(self, env, config)
self.logger = get_logger(config.log_path)
self.avg_reward = 0
self.progress = Progbar(target=self.config.nsteps_train)