Skip to content
This repository was archived by the owner on Mar 14, 2025. It is now read-only.

Commit 8dcd18c

Browse files
committed
add missing json import
1 parent 8f1711e commit 8dcd18c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

network/dump_network.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,8 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
import json
16+
1517
# Reference: https://devtalk.nvidia.com/default/topic/1064669/tensorrt/troubleshooting-suggestions-for-onnx-v-tensorrt-discrepancies/post/5392296/#5392296
1618
def dump_network(network, filename):
1719
"""Dumps TensorRT parsed network attributes to a JSON file.

0 commit comments

Comments
 (0)