helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-11-07 17:27:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(ml): batch axis not being added for recognition model (#12588)

Browse source 
* fix has_batch_axis

* fix typing
This commit is contained in:
Mert 2024-09-11 23:21:12 -04:00 • committed by GitHub
parent fa095c3ca0
commit 22dc9bcebb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 1 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
machine-learning/app/sessions/__init__.py
View file

@ -1,5 +0,0 @@
from app.schemas import ModelSession
def has_batch_axis(session: ModelSession) -> bool:
return not isinstance(session.get_inputs()[0].shape[0], int) or session.get_inputs()[0].shape[0] < 0