search

TrojanXu / yolov5-tensorrt

A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5
Apache License 2.0
190 stars 46 forks source link

Wont work with V3.0 or JetPack 4.4 #37

Open GuiSteffen opened 4 years ago

GuiSteffen commented 4 years ago

The code is not yet available to work on Jetpack 4.4.

The current version requires PYTorch 1.4.0 to be installed, altough the Jetpack 4.4 production dont support Pytorch < 1.6.0.

Anybody using Jetpack 4.4 are currently in a loop hole. Follows the error (PyTorch Incompatibility)

Does anybody has a solution to this? Will we have an implementation of TRT for Pytorch 1.6?

torch.nn.modules.module.ModuleAttributeError: 'BatchNorm2d' object has no attribute '_non_persistent_buffers_set'

TrojanXu commented 4 years ago

Updated with tag v3.0