Commit 1f170b13 authored by Benjamin Beyret's avatar Benjamin Beyret
Browse files

agent.py: revert to 84x84 for evaluation

parent 3c739478
......@@ -9,14 +9,13 @@ class Agent(object):
"""
Load your agent here and initialize anything needed
"""
# Load the configuration and model using ABSOLUTE PATHS
self.configuration_file = '/aaio/data/trainer_config.yaml'
self.model_path = '/aaio/data/1-Food/Learner'
self.brain = BrainParameters(brain_name='Learner',
camera_resolutions=[
{'height': self.resolution, 'width': self.resolution, 'blackAndWhite': False}],
camera_resolutions=[{'height': 84, 'width': 84, 'blackAndWhite': False}],
num_stacked_vector_observations=1,
vector_action_descriptions=['', ''],
vector_action_space_size=[3, 3],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment