models_mnist/generator.py [113:133]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    self.outputs = outputs
    self.inputs = {ii: inputs[ii] for ii in used_inputs}
  #------------------------------------------------------------

  # setters and getters
  def get_outputs(self):
    return self.outputs

  def get_inputs(self):
    return self.inputs
  #------------------------------------------------------------

  # produce feed dict
  def produce_feed_dict(self, batch, prev_output=None):
    feed_dict = {}
    feed_dict[self.inputs['ques']] = batch['ques']
    feed_dict[self.inputs['ques_len']] = batch['ques_len']

    # add program
    if 'prog' in self.params['model']:
      feed_dict[self.inputs['prog_gt']] = batch['gt_layout']
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



models_vd/generator.py [220:240]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    self.outputs = outputs
    self.inputs = {ii: inputs[ii] for ii in used_inputs}
  #------------------------------------------------------------

  # setters and getters
  def get_outputs(self):
    return self.outputs

  def get_inputs(self):
    return self.inputs
  #------------------------------------------------------------

  # produce feed dict
  def produce_feed_dict(self, batch, prev_output=None):
    feed_dict = {}
    feed_dict[self.inputs['ques']] = batch['ques']
    feed_dict[self.inputs['ques_len']] = batch['ques_len']

    # add program
    if 'prog' in self.params['model']:
      feed_dict[self.inputs['prog_gt']] = batch['gt_layout']
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -