jch-wang
jch-wang
> 最后一层在pytorch是adaptive pooling，这边用的固定kernel，需要根据input size改一下 那就是说tensorrx中的输出是没有问题的？ pytorch的实现里需要在哪个文件里改kernel呀？请原谅我太菜...
意思是修改tensorrtx/squeezenet/squeezenet.cpp中第137行的： `IPoolingLayer* pool3 = network->addPooling(*relu2->getOutput(0), PoolingType::kAVERAGE, DimsHW{14, 14});` DimsHW的大小吗？应该修改为多少？....脑子突然不在线....
> 对，pool3输出是1x1，所以要根据前一层的shape改 感谢，已解决~
又来打扰了~ 我想将squeezenet输出由原来的1000维改成512维，squeezenet.cpp中的输出维度可以直接修改OUTPUT_SIZE参数，请问在squeezenet.py中的，我应该怎么修改呢？
> File does not exist : ../configs/yolov4-kFLOAT-batch4.engine > Building the TensorRT Engine... > Building complete! > Serializing the TensorRT Engine... > Killed 补充：也报过这个错误： wjc@wjc-desktop:~/yolo-tensorrt/build$ ./yolo-trt File does not exist :...
> config_v4.inference_precison = FP32; > 改这个。fp16 int8 晓得了，感谢指点，我去尝试。
> support all layers,no need write any layer. Do you mean tensorrt 7?