search

google / automl

Google Brain AutoML
Apache License 2.0
6.23k stars 1.45k forks source link

cannot train with ciou enabled #543

Closed leedewdew closed 3 years ago

leedewdew commented 4 years ago

When I train with different iou loss, diou and giou works fine, but ciou does not. It blocks the training telling "Saving checkpoints for 0 into ..." and no more progess.

zishanahmed08 commented 4 years ago

i face the same issue

zpillio commented 4 years ago

Maybe it's related (maybe not) but for me 'iou_utils_test.py' fails in 'ciou' cases with tensorflow version < 2.0: "TypeError: If using @custom_gradient with a function that uses variables, the enclosing variable scope must have use_resource=True."

leedewdew commented 4 years ago

Maybe it's related (maybe not) but for me 'iou_utils_test.py' fails in 'ciou' cases with tensorflow version < 2.0: "TypeError: If using @custom_gradient with a function that uses variables, the enclosing variable scope must have use_resource=True."

Yes, tf1.0 fails with ciou but tf2.0 works.