return ret
def _dense_block_end_node(self, layer_id):
return self._block_end_node(layer_id, Constant.DENSE_BLOCK_DISTANCE)
def _conv_block_end_node(self, layer_id):
Get the input node ID of the last layer in the block by layer ID.
Return the input node ID of the last layer in the convolutional block.
After Change
return ret
def _dense_block_end_node(self, layer_id):
return self.layer_id_to_input_node_ids[layer_id][0]
def _conv_block_end_node(self, layer_id):
Get the input node ID of the last layer in the block by layer ID.
Return the input node ID of the last layer in the convolutional block.
