Skip to content

Instantly share code, notes, and snippets.

This file has been truncated, but you can view the full file.
[08/29/2022-22:12:27] [TRT] [I] [MemUsageChange] Init CUDA: CPU +319, GPU +0, now: CPU 3610, GPU 2730 (MiB)
[08/29/2022-22:12:28] [TRT] [I] [MemUsageChange] Init builder kernel library: CPU +327, GPU +106, now: CPU 3956, GPU 2836 (MiB)
[08/29/2022-22:12:28] [TRT] [I] [MemUsageChange] Init CUDA: CPU +0, GPU +0, now: CPU 3956, GPU 2836 (MiB)
[08/29/2022-22:12:28] [TRT] [V] Registered plugin creator - ::GridAnchor_TRT version 1
[08/29/2022-22:12:28] [TRT] [V] Registered plugin creator - ::GridAnchorRect_TRT version 1
[08/29/2022-22:12:28] [TRT] [V] Registered plugin creator - ::NMS_TRT version 1
[08/29/2022-22:12:28] [TRT] [V] Registered plugin creator - ::Reorg_TRT version 1
[08/29/2022-22:12:28] [TRT] [V] Registered plugin creator - ::Region_TRT version 1
[08/29/2022-22:12:28] [TRT] [V] Registered plugin creator - ::Clip_TRT version 1
[08/29/2022-22:12:28] [TRT] [V] Registered plugin creator - ::LReLU_TRT version 1
@jakepoz
jakepoz / torchscript_export.py
Created June 26, 2020 23:47
Exports a YoloV5 model as torchscript
"""Exports a pytorch *.pt model to *.onnx format
Usage:
$ export PYTHONPATH="$PWD" && python models/onnx_export.py --weights ./weights/yolov5s.pt --img 640 --batch 1
"""
import argparse
import onnx
if (Windows.Foundation.Metadata.ApiInformation.IsTypePresent("Windows.UI.ViewManagement.StatusBar"))
{
StatusBar statusBar = Windows.UI.ViewManagement.StatusBar.GetForCurrentView();
// Hide the status bar
statusBar.HideAsync();
}