Keras replacing input layer

前端 未结 6 2114
清酒与你
清酒与你 2020-11-30 07:29

The code that I have (that I can\'t change) uses the Resnet with my_input_tensor as the input_tensor.

model1 = keras.applications.resnet50.ResNe         


        
6条回答
  •  日久生厌
    2020-11-30 08:07

    For tf.keras in Tensorflow 2, using tfsurgeon did not work as I had custom layers.

    What did work was changing layer._batch_input_size (note the underscore)

    Hacky, but got the job done.

提交回复
热议问题