Skip to content

Instantly share code, notes, and snippets.

@rayheberer
Created August 17, 2018 00:23
Show Gist options
  • Save rayheberer/23f1e4938d8822bfda9714623412a60f to your computer and use it in GitHub Desktop.
Save rayheberer/23f1e4938d8822bfda9714623412a60f to your computer and use it in GitHub Desktop.
class AtariNet(object):
# ...
# ...
# ...
def _build(self):
# ...
# ...
self.screen_features = tf.placeholder(
tf.int32,
[None, len(SCREEN_FEATURES), *self.screen_dimensions],
name="screen_features")
self.minimap_features = tf.placeholder(
tf.int32,
[None, len(MINIMAP_FEATURES), *self.minimap_dimensions],
name="minimap_features")
self.flat_features = tf.placeholder(
tf.float32,
[None, len(features.Player)],
name="flat_features")
# ...
# ...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment