search

talebolano / TensorRT-solo-python

TensorRT for SOLO(use python)
Other
26 stars 6 forks source link

light solo may change this 256? #1

Closed jiangfeng94 closed 3 years ago

jiangfeng94 commented 3 years ago

kernel_pred_list = [ outs[1][i].squeeze(0).permute(1, 2, 0).view(-1, 256) for i in range(5)] # if use light solo may change this 256

i dont know how to change it.