models_low_rank.py 文件源码

def _update_deploy_net(self):
        """ Update deploy net. """
        self._deploy_net = caffe.NetSpec()
        data = self._add_input_layers(self._deploy_net, deploy=True)
        bottom_dict = self._add_intermediate_layers(self._deploy_net, data, deploy=True)
        self._add_output_layers(self._deploy_net, bottom_dict, deploy=True)
        # add input definition strings.
        self._deploy_str='input: {}\ninput_dim: {}\ninput_dim: {}\ninput_dim: {}\ninput_dim: {}'.\
            format('"'+self.io.data_name+'"', 1, 3, 224, 224)