diff --git a/examples/efficientnet.py b/examples/efficientnet.py index af6bf8acdc..41e871be70 100644 --- a/examples/efficientnet.py +++ b/examples/efficientnet.py @@ -45,7 +45,7 @@ def infer(model, img): """ # run the net - out = _infer(model, Tensor(img)).numpy() + out = _infer(model, Tensor(img.astype("float32"))).numpy() # if you want to look at the outputs """