Merge branch 'master' of ssh://github.com/kristjankorjus/Replicating-…

…DeepMind
kristjankorjus · Dec 17, 2014 · 9313e21 · 9313e21
2 parents b2b9413 + 1ab0d1d
commit 9313e21
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/libraries/ale/makefile.unix b/libraries/ale/makefile.unix
@@ -4,7 +4,7 @@
 
 srcdir      ?= .
 # Set this to 1 to enable SDL and display_screen
-USE_SDL     := 0
+USE_SDL     := 1
 # Set this to 1 to enable the RLGlue interface
 USE_RLGLUE  := 0
 DEFINES     := -DRLGENV_NOMAINLOOP

diff --git a/libraries/ale/roms/seaquest.bin b/libraries/ale/roms/seaquest.bin
diff --git a/src/ale/ale.py b/src/ale/ale.py
@@ -21,7 +21,7 @@ class ALE:
     fout = ""
     preprocessor = None
 
-    def __init__(self,  memory, display_screen="true", skip_frames=4, game_ROM='../libraries/ale/roms/breakout.bin'):
+    def __init__(self,  memory, display_screen, skip_frames, game_ROM):
         """
         Initialize ALE class. Creates the FIFO pipes, launches ./ale and does the "handshake" phase of communication
 

diff --git a/src/main.py b/src/main.py
@@ -59,7 +59,7 @@ class Main:
 
     def __init__(self):
         self.memory = MemoryD(self.memory_size)
-        self.ale = ALE(self.memory)
+        self.ale = ALE(self.memory, display_screen="true", skip_frames=4, game_ROM='../libraries/ale/roms/breakout.bin')
         self.nnet = NeuralNet(self.state_size, self.number_of_actions, "ai/deepmind-layers.cfg", "ai/deepmind-params.cfg", "layer4")
 
     def compute_epsilon(self, frames_played):