Skip to content

fix: align input tensors to model device in predict instead of cpu#94

Merged
meilame-tayebjee merged 2 commits into
mainfrom
fix/predict-device-mismatch
Jun 10, 2026
Merged

fix: align input tensors to model device in predict instead of cpu#94
meilame-tayebjee merged 2 commits into
mainfrom
fix/predict-device-mismatch

Add device parameter to model evaluation

593364c
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar