tensorrtx
tensorrtx copied to clipboard
Implementation of popular deep learning networks with TensorRT network definition API
## Env - GPU, e.g. V100, RTX2080, TX2, Xavier NX, Nano, etc. - OS, e.g. Ubuntu16.04, Win10, etc. - Cuda version - TensorRT version ## About this repo - which...
jetson@yahboom:~/yolov5$ python3 gen_wts.py -w new.pt Traceback (most recent call last): File "gen_wts.py", line 33, in model = torch.load(pt_file, map_location=device) File "/home/jetson/.local/lib/python3.6/site-packages/torch/serialization.py", line 592, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)...
## Env ## About this repo ./yolov5_det -s bestss.wts bestss.engine s [W] [TRT] The implicit batch dimension mode has been deprecated. Please create the network with NetworkDefinitionCreationFlag::kEXPLICIT_BATCH flag whenever possible....
## Env - GPU, NVIDIA GeForce GTX 1650 - OS, Ubuntu18.04 - Cuda version:None - TensorRT version:yolov5-v5.0 ## About this repo - which branch/tag/commit are you using? tensorrtx-yolov5-v5.0 - which...