Pin `keras` version to 3.5.0

brainglobe / cellfinder

Automated 3D cell detection in very large images

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

181 stars 39 forks source link

Closed IgorTatarnikov closed 1 month ago

IgorTatarnikov commented 1 month ago

Currently, training enters an infinite loop at the end of an epoch.

This only happens with keras==3.6.0. Temporarily pinning to 3.5.0.