gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-03-12 12:35:49 +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

5
rl_coach/architectures/mxnet_components/embedders/__init__.py
View File

@@ -1,4 +1,7 @@
from .image_embedder import ImageEmbedder
from .tensor_embedder import TensorEmbedder
from .vector_embedder import VectorEmbedder
__all__ = ['ImageEmbedder', 'VectorEmbedder']
__all__ = ['ImageEmbedder',
'TensorEmbedder',
'VectorEmbedder']