📌  相关文章
📜  没有@tf.function OOM - Python 代码示例

📅  最后修改于: 2022-03-11 14:46:50.397000             🧑  作者: Mango

代码示例1
def train_step(self, x, step, grad_clip=True, clip_value=1.0):
    print('tracing', x, step, grad_clip, clip_value)
    with tf.name_scope("input_data"):
      ...