Не удается преобразовать ошибку типа в theano.function() для TensorType (float64, matrix) в Type TensorType (float32, matrix)

Question

Я хочу запустить код через pylearn2 с магистралью Theano. updatesэто OrderedDictи большинство его значений имеют тип float64, а два из них имеют тип float32. Когда я хочу запустить код, и он достигает этой строки

      train_fn = theano.function([input, target, LR], loss, updates=updates)

возникла эта ошибка.

      Traceback (most recent call last):
  File "/usr/local/lib/python3.9/site-packages/theano/compile/pfunc.py", line 192, in rebuild_collect_shared
    update_val = store_into.type.filter_variable(update_val,
  File "/usr/local/lib/python3.9/site-packages/theano/tensor/type.py", line 229, in filter_variable
raise TypeError(
TypeError: Cannot convert Type TensorType(float64, matrix) (of Variable Elemwise{add,no_inplace}.0) into Type TensorType(float32, matrix). 
You can try to manually convert Elemwise{add,no_inplace}.0 into a TensorType(float32, matrix).

Он предлагает конвертировать вручную.