models_low_rank.py 文件源码

def _update_trainval_net(self):
        """ Update trainval net. """
        self._train_net = caffe.NetSpec()
        self._val_net = caffe.NetSpec()
        in_nets = {'train': self._train_net, 'val': self._val_net}
        data = self._add_input_layers(in_nets, deploy=False)
        bottom_dict = self._add_intermediate_layers(self._train_net, data, deploy=False)
        self._add_output_layers(self._train_net, bottom_dict, deploy=False)