ARNING:tensorflow:Worker c7428b79-0a28-40ed-a220-10946fd0c4af run 00001: Exception:
Traceback (most recent call last):
  File "/homes/vl304/Documents/orthnn/planet/planet/training/running.py", line 201, in __iter__
    for value in self._process_fn(self._logdir, *args):
  File "/homes/vl304/Documents/orthnn/planet/planet/scripts/train.py", line 73, in process
    training.define_model, dataset, logdir, config):
  File "/homes/vl304/Documents/orthnn/planet/planet/training/utility.py", line 166, in train
    score, summary, cleanups = model_fn(data, trainer, config)
  File "/homes/vl304/Documents/orthnn/planet/planet/training/define_model.py", line 95, in define_model
    name='should_collect_' + name)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 2097, in cond
    orig_res_t, res_t = context_t.BuildCondBranch(true_fn)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 1941, in BuildCondBranch
    original_result = fn()
  File "/homes/vl304/Documents/orthnn/planet/planet/training/utility.py", line 305, in simulate_episodes
    expensive_summaries, gif_summary, name=name)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 45, in simulate
    isolate_envs=isolate_envs)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 86, in collect_rollouts
    initializer, parallel_iterations=1)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/functional_ops.py", line 724, in scan
    maximum_iterations=n)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3556, in while_loop
    return_same_structure)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3087, in BuildLoop
    pred, body, original_loop_vars, loop_vars, shape_invariants)
  File "/homes/vl304/.local/lib/ARNING:tensorflow:Worker c7428b79-0a28-40ed-a220-10946fd0c4af run 00001: Exception:
Traceback (most recent call last):
  File "/homes/vl304/Documents/orthnn/planet/planet/training/running.py", line 201, in __iter__
    for value in self._process_fn(self._logdir, *args):
  File "/homes/vl304/Documents/orthnn/planet/planet/scripts/train.py", line 73, in process
    training.define_model, dataset, logdir, config):
  File "/homes/vl304/Documents/orthnn/planet/planet/training/utility.py", line 166, in train
    score, summary, cleanups = model_fn(data, trainer, config)
  File "/homes/vl304/Documents/orthnn/planet/planet/training/define_model.py", line 95, in define_model
    name='should_collect_' + name)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 2097, in cond
    orig_res_t, res_t = context_t.BuildCondBranch(true_fn)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 1941, in BuildCondBranch
    original_result = fn()
  File "/homes/vl304/Documents/orthnn/planet/planet/training/utility.py", line 305, in simulate_episodes
    expensive_summaries, gif_summary, name=name)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 45, in simulate
    isolate_envs=isolate_envs)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 86, in collect_rollouts
    initializer, parallel_iterations=1)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/functional_ops.py", line 724, in scan
    maximum_iterations=n)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3556, in while_loop
    return_same_structure)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3087, in BuildLoop
    pred, body, original_loop_vars, loop_vars, shape_invariants)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3022, in _BuildLoop
    body_result = body(*packed_vars_for_body)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3525, in <lambda>
    body = lambda i, lv: (i + 1, orig_body(*lv))
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/functional_ops.py", line 703, in compute
    a_out = fn(packed_a, packed_elems)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 71, in simulate_fn
    reset=tf.equal(step, 0))
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 233, in simulate_step
    step, score, length = _define_step()
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 162, in _define_step
    action, step_summary = algo.perform(agent_indices, prevob)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/mpc_agent.py", line 67, in perform
    _, state = self._cell((embedded, prev_action, use_obs), state)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/rnn_cell_impl.py", line 234, in __call__
    return super(RNNCell, self).__call__(inputs, state, precomputed_data=precomputed_data)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/layers/base.py", line 530, in __call__
    outputs = super(Layer, self).__call__(inputs, *args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/keras/engine/base_layer.py", line 554, in __call__
    outputs = self.call(inputs, *args, **kwargs)
  File "/homes/vl304/Documents/orthnn/planet/planet/models/base.py", line 60, in call
    prior = self._transition_tpl(prev_state, prev_action, zero_obs, transition_matrix=precomputed_data)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/template.py", line 360, in __call__
    return self._call_func(args, kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/template.py", line 306, in _call_func
    result = self._func(*args, **kwargs)
  File "/homes/vl304/Documents/orthnn/planet/planet/models/orssm.py", line 106, in _transition
    belief, rnn_state = self._cell(hidden, prev_state['rnn_state'], precomputed_data=transition_matrix)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/rnn_cell_impl.py", line 371, in __call__
    *args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/layers/base.py", line 530, in __call__
    outputs = super(Layer, self).__call__(inputs, *args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/keras/engine/base_layer.py", line 554, in __call__
    outputs = self.call(inputs, *args, **kwargs)
  File "/homes/vl304/Documents/orthnn/planet/planet/models/cwy_cell.py", line 42, in call
    print('SHAPES!!!!!!!!!!!!!!!!!!', h_prev.shape, precomputed_data.shape)
AttributeError: 'NoneType' object has no attribute 'shape'
python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3022, in _BuildLoop
    body_result = body(*packed_vars_for_body)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/control_flow_ops.py", line 3525, in <lambda>
    body = lambda i, lv: (i + 1, orig_body(*lv))
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/functional_ops.py", line 703, in compute
    a_out = fn(packed_a, packed_elems)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 71, in simulate_fn
    reset=tf.equal(step, 0))
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 233, in simulate_step
    step, score, length = _define_step()
  File "/homes/vl304/Documents/orthnn/planet/planet/control/simulate.py", line 162, in _define_step
    action, step_summary = algo.perform(agent_indices, prevob)
  File "/homes/vl304/Documents/orthnn/planet/planet/control/mpc_agent.py", line 67, in perform
    _, state = self._cell((embedded, prev_action, use_obs), state)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/rnn_cell_impl.py", line 234, in __call__
    return super(RNNCell, self).__call__(inputs, state, precomputed_data=precomputed_data)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/layers/base.py", line 530, in __call__
    outputs = super(Layer, self).__call__(inputs, *args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/keras/engine/base_layer.py", line 554, in __call__
    outputs = self.call(inputs, *args, **kwargs)
  File "/homes/vl304/Documents/orthnn/planet/planet/models/base.py", line 60, in call
    prior = self._transition_tpl(prev_state, prev_action, zero_obs, transition_matrix=precomputed_data)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/template.py", line 360, in __call__
    return self._call_func(args, kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/template.py", line 306, in _call_func
    result = self._func(*args, **kwargs)
  File "/homes/vl304/Documents/orthnn/planet/planet/models/orssm.py", line 106, in _transition
    belief, rnn_state = self._cell(hidden, prev_state['rnn_state'], precomputed_data=transition_matrix)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/ops/rnn_cell_impl.py", line 371, in __call__
    *args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/layers/base.py", line 530, in __call__
    outputs = super(Layer, self).__call__(inputs, *args, **kwargs)
  File "/homes/vl304/.local/lib/python3.5/site-packages/tensorflow/python/keras/engine/base_layer.py", line 554, in __call__
    outputs = self.call(inputs, *args, **kwargs)
  File "/homes/vl304/Documents/orthnn/planet/planet/models/cwy_cell.py", line 42, in call
    print('SHAPES!!!!!!!!!!!!!!!!!!', h_prev.shape, precomputed_data.shape)
AttributeError: 'NoneType' object has no attribute 'shape'