gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

added a simple progress bar implementation

Browse Source
This commit is contained in:
itaicaspi-intel
2018-09-13 14:21:38 +03:00
parent fa79d8d365
commit 607ef17431
3 changed files with 34 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/environments/carla_environment.py
View File

@@ -378,7 +378,7 @@ class CarlaEnvironment(Environment):
self.control.brake = np.abs(np.clip(action[0], -1, 0))
# prevent braking
if not self.allow_braking:
if not self.allow_braking or self.control.brake < 0.1:
self.control.brake = 0
# prevent over speeding

19
rl_coach/memories/non_episodic/experience_replay.py
View File

@@ -22,8 +22,9 @@ import time
import numpy as np
from rl_coach.core_types import Transition
from rl_coach.logger import screen
from rl_coach.memories.memory import Memory, MemoryGranularity, MemoryParameters
from rl_coach.utils import ReaderWriterLock
from rl_coach.utils import ReaderWriterLock, ProgressBar
class ExperienceReplayParameters(MemoryParameters):
@@ -239,15 +240,17 @@ class ExperienceReplay(Memory):
with open(file_path, 'rb') as file:
transitions = pickle.load(file)
num_transitions = len(transitions)
start_time = time.time()
if num_transitions > self.max_size[1]:
screen.warning("Warning! The number of transition to load into the replay buffer ({}) is "
"bigger than the max size of the replay buffer ({}). The excessive transitions will "
"not be stored.".format(num_transitions, self.max_size[1]))
progress_bar = ProgressBar(num_transitions)
for transition_idx, transition in enumerate(transitions):
self.store(transition)
# print progress
if transition_idx % 100 == 0:
percentage = int((100 * transition_idx) / num_transitions)
sys.stdout.write("\rProgress: ({}/{})".format(transition_idx, num_transitions))
sys.stdout.write(' Time (sec): {}'.format(round(time.time() - start_time, 2)))
sys.stdout.write(' {}%|{}{}| '.format(percentage, '#' * int(percentage / 10),
' ' * (10 - int(percentage / 10))))
sys.stdout.flush()
progress_bar.update(transition_idx)
progress_bar.close()

21
rl_coach/utils.py
View File

@@ -20,6 +20,7 @@ import inspect
import json
import os
import signal
import sys
import threading
import time
from multiprocessing import Manager
@@ -549,3 +550,23 @@ class ReaderWriterLock(object):
self.num_readers_lock.acquire()
self.num_readers -= 1
self.num_readers_lock.release()
class ProgressBar(object):
def __init__(self, max_value):
self.start_time = time.time()
self.max_value = max_value
self.current_value = 0
def update(self, current_value):
self.current_value = current_value
percentage = int((100 * current_value) / self.max_value)
sys.stdout.write("\rProgress: ({}/{}) Time: {} sec {}%|{}{}| "
.format(current_value, self.max_value,
round(time.time() - self.start_time, 2),
percentage, '#' * int(percentage / 10),
' ' * (10 - int(percentage / 10))))
sys.stdout.flush()
def close(self):
print("")