onnxruntime
d5606cd7 - Introducing customizable input names for loss in generate_artifacts. (#19705)

Commit
1 year ago
Introducing customizable input names for loss in generate_artifacts. (#19705) # loss function extra inputs. Currently, the loss functions in onnxblock expect exactly two inputs in their build method. Occasionally, models may pass additional inputs, causing the build function to fail. To solve this issue, we can let users pass a list of loss input names to be used in the loss function.
Author
Parents
Loading