(self)
| 277 | |
| 278 | # compute loss |
| 279 | def _create_loss(self): |
| 280 | self._logits = tf.squeeze(self._logits) |
| 281 | self.loss = tf.losses.sigmoid_cross_entropy( |
| 282 | self._label, |
| 283 | self._logits, |
| 284 | scope='loss', |
| 285 | reduction=tf.losses.Reduction.SUM_OVER_BATCH_SIZE) |
| 286 | tf.summary.scalar('loss', self.loss) |
| 287 | |
| 288 | # define optimizer and generate train_op |
| 289 | def _create_optimizer(self): |