Skip to content

Tag: tensorflow

Import onnx models to tensorflow2.x?

I created a modified lenet model using tensorflow that looks like this: When I finish training I save the model using tf.keras.models.save_model : Then I transform this model into onnx format using “tf2onnx” module: I want a method that can retrieve the same model into tensorflow2.x. I tried to us…

How to run scipy’s BFGS on GPU

I’d like to run scipy implementation of BFGS optimization algorithm on GPU and scipy seems not to support GPUs. The target function which I want to run on GPU is the following one which is part of the implementation of this repository: I know there is Tensorflow Probablity implementation of BFGS, but I …