experiencor / keras-yolo2

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).
MIT License
1.73k stars 787 forks source link

Doesn't work anymore #455

Open Zumbalamambo opened 4 years ago

Zumbalamambo commented 4 years ago

It fails when it accessed GPU. Even after reducing the batch size in colab, it still fails.

If I try to train in CPU, it throws the following error,

Epoch 1/4
2020-06-06 23:16:08.769001: W tensorflow/core/framework/op_kernel.cc:1753] OP_REQUIRES failed at resource_variable_ops.cc:541 : Not found: Resource localhost/_AnonymousVar117/class tensorflow::Var does not exist.
2020-06-06 23:16:09.299689: W tensorflow/core/framework/op_kernel.cc:1753] OP_REQUIRES failed at resource_variable_ops.cc:541 : Not found: Resource localhost/_AnonymousVar118/class tensorflow::Var does not exist.

I'm using tensorflow version 2.3