def train_op(self) -> tf.Operation:
"""
Returns an operation for performing a single training step.
Returns
-------
tf.Operation
An operation for performing a single training step
"""
return self.graph.get_collection("train_op")[0]
评论列表
文章目录