I’m using the Clara train v3 docker and used the custom model code provided in the docs : Bring your own components. Training works fine but when trying to do inference from the exported model, I get the following error :
KeyError: “The name ‘NV_IS_TRAINING:0’ refers to a Tensor which does not exist. The operation, ‘NV_IS_TRAINING’, does not exist in the graph.”
I’ve done inference on existing models and those seem to work fine, I’m guessing something should be included in the code when writing a custom model for inference?