New classification model/tf version is much slower than main

SINTEF / pyopia

Python Ocean Particle Image Analysis

https://pyopia.readthedocs.io

BSD 3-Clause "New" or "Revised" License

11 stars 5 forks source link

Closed nepstad closed 2 months ago

nepstad commented 2 months ago

The classification step on dev-2.0.0 branch is much slower than on main (a factor of 6x in some cases).

It seems to be related to calling model.predict for each individual particle, ref: https://github.com/tensorflow/tensorflow/issues/40261