This commit is contained in:
Christoffer Müller Madsen 2018-04-19 16:04:49 +02:00
parent 66589dfde3
commit 1062b72bda

View File

@ -164,7 +164,7 @@ class Network:
# Restore trained episode count for model # Restore trained episode count for model
episode_count_path = os.path.join(self.checkpoint_path, "episodes_trained") episode_count_path = os.path.join(self.checkpoint_path, "episodes_trained")
p if os.path.isfile(episode_count_path): if os.path.isfile(episode_count_path):
with open(episode_count_path, 'r') as f: with open(episode_count_path, 'r') as f:
self.config['start_episode'] = int(f.read()) self.config['start_episode'] = int(f.read())