add missing json import

rmccorm4 · Nov 25, 2019 · 8dcd18c · 8dcd18c
1 parent 8f1711e
commit 8dcd18c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/network/dump_network.py b/network/dump_network.py
@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import json
+
 # Reference: https://devtalk.nvidia.com/default/topic/1064669/tensorrt/troubleshooting-suggestions-for-onnx-v-tensorrt-discrepancies/post/5392296/#5392296
 def dump_network(network, filename):
     """Dumps TensorRT parsed network attributes to a JSON file.