Try decorating it directly with @tf.function.

Visual-Behavior / detr-tensorflow

Tensorflow implementation of DETR : Object Detection with Transformers

MIT License

169 stars 53 forks source link

Try decorating it directly with @tf.function. #32

Open atp1988 opened 3 years ago

atp1988 commented 3 years ago

How could I pass from this Error? OperatorNotAllowedInGraphError: using a tf.Tensor as a Python bool is not allowed: AutoGraph is disabled in this function. Try decorating it directly with @tf.function.

when I train the code through train_coco.py file, this error happend? is anyone could help me?