def build(self, net):
self.logits = slim.fully_connected(
net, self.num_classes, activation_fn=None)
self.prediction = slim.softmax(self.logits)
微信
公众号
扫码关注公众号