2cacee3efdf68a041330d2059c7cdb41dcc7df72,calamari_ocr/ocr/backends/tensorflow_backend/tensorflow_model.py,TensorflowModel,load,#Any#Any#,18

Before Change

            // TODO: Crash if cudnn is loaded
            // Workaround create new graph and load weights
            print(e)
            print("Attempting a workaround")

            model = TensorflowModel.from_proto(network_proto)
            with model.graph.as_default() as g:
                saver = tf.train.Saver()

After Change

                    saver.restore(model.session, restore)
                except tf.errors.NotFoundError as e:
                    print("Attempting workaround: only loading trainable variables")
                    saver = tf.train.Saver(tf.trainable_variables())
                    saver.restore(model.session, restore)

            return model

Instances