Without @tf.function OOM

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"):
      ...

Source: github.com

Similar codes

2 votes380 views4 answers
How to create dictionary in python from csv
Python
12 votes451 views4 answers
Flask debug
Python Flask
8 votes339 views1 answer
Python transfer file
Python
9 votes299 views3 answers
A value is trying to be set on a copy of a slice from a DataFrame.
Python
0 vote333 views1 answer
Django dump previous migrations
Python
3 votes429 views1 answer
Django admin prefetch_related
Python Django
12 votes344 views2 answers
Python pip version check
Python
2 votes415 views1 answer
Switching keys and values in a dictionary in python [duplicate]
Python
0 vote351 views1 answer
Find where df series is null and print
Python
1 vote366 views1 answer
Python add current directory to import path
Python