KittiSeg icon indicating copy to clipboard operation
KittiSeg copied to clipboard

Train and save model without dropout layers

Open kmonachopoulos opened this issue 8 years ago • 0 comments

Is there a way to adjust the architecture of the model in a way that after training can be saved without the dropout layers? After saving it to protobuf format it is nearly impossible to remove and adjust the layers, even optimize_for_inference or transform_graph scripts cannot remove them. It is a common problem.

kmonachopoulos avatar Sep 03 '17 19:09 kmonachopoulos