Fixes a bug where we were trying to access weight tensor for inference. This...
Fixes a bug where we were trying to access weight tensor for inference. This broke export/inference for any classification model using weights. + Added a test. Change: 146697049
Loading
Please sign in to comment