Training a models throws "could not create a primitive descriptor for an LSTM forward propagation primitive"

mittagessen / kraken

OCR engine for all the languages

http://kraken.re

Apache License 2.0

750 stars 131 forks source link

Training a models throws "could not create a primitive descriptor for an LSTM forward propagation primitive" #610

Closed PonteIneptique closed 5 months ago

PonteIneptique commented 5 months ago

ketos train --augment -f binary ./dataset.arrow -B 16 -i ~/Downloads/catmus-print-fondue-large.mlmodel --lrate 0.0001 -u NFKD --resize resize --fixed-splits --precision bf16 --lag 5 runs

RuntimeError: could not create a primitive descriptor for an LSTM forward propagation primitive

Fresh install from this morning

PonteIneptique commented 5 months ago

OK, forgot --device cuda:0...