search

Leonardo-Blanger / detr_tensorflow

A Tensorflow implementation of the DETR object detection architecture.
MIT License
62 stars 15 forks source link

Model cannot be saved #5

Open ashishverma818 opened 3 years ago

ashishverma818 commented 3 years ago

Trying to save the model using tf.keras.models.save_model leads to a TypeError:

File "detr_tf/networks/transformer.py", line 211, in call key_mem = memory + pos_encoding TypeError: unsupported operand type(s) for +: 'NoneType' and 'NoneType'

pranavjadhav001 commented 2 years ago

Better way is to use save_weights and use load_weights with the architecture,the above issue comes with a custom built model/layer