Skip to content

Use Keras (with Torch by default) - #338

Merged
cbrnr merged 2 commits into
mainfrom
keras3
Aug 7, 2026
Merged

cbrnr merged 2 commits into
mainfrom
keras3

Conversation

@cbrnr

@cbrnr cbrnr commented Aug 7, 2026

Copy link
Copy Markdown
Owner

Fixes #259. I think we don't have to move away from Keras, as this is now a source package anyway (so we don't have the problem that binary wheels lag behind Python releases). The default backend is Torch, but in principle Tensorflow still works, and we don't have to convert any weights or retrain the shipped classifiers.

@cbrnr
cbrnr merged commit 6e10222 into main Aug 7, 2026
10 checks passed
@cbrnr
cbrnr deleted the keras3 branch August 7, 2026 13:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Switch from TensorFlow (Keras) to PyTorch

1 participant