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

Add tensor input type for arbitrary dimensional observation (#125)

Browse Source 
* Allow arbitrary dimensional observation (non vector or image)
* Added creating PlanarMapsObservationSpace to GymEnvironment when number of channels is not 1 or 3
This commit is contained in:
Sina Afrooze
2018-11-19 06:41:12 -08:00
committed by Gal Leibovich
parent 7ba1a4393f
commit 67a90ee87e
10 changed files with 194 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/architectures/embedder_parameters.py
View File

@@ -30,9 +30,9 @@ class InputEmbedderParameters(NetworkComponentParameters):
self.dropout_rate = dropout_rate
if input_rescaling is None:
input_rescaling = {'image': 255.0, 'vector': 1.0}
input_rescaling = {'image': 255.0, 'vector': 1.0, 'tensor': 1.0}
if input_offset is None:
input_offset = {'image': 0.0, 'vector': 0.0}
input_offset = {'image': 0.0, 'vector': 0.0, 'tensor': 0.0}
self.input_rescaling = input_rescaling
self.input_offset = input_offset