Extracting the graph

[Wonder1905]

Since it's tf2, it's much harder for me to extract the forward graph (also the backward), can you help us with that?

[akanyaani]

Hi @BattashB,

Sorry for my late reply, Can you elaborate the issue?

Thanks

[ray1422]

how about use tf.saved_model ?

[akanyaani]

Hi @ray1422

#you can run this it will give .PB format of model.

tf.saved_model.save(gpt2_model, model_path)

#Run this line if you get some signature related error.

input_signature = #Your input tesnor signature
model._set_inputs(input_signature)