xxd1999

Train erro

4

您好，我在用yolov3-tiny训练自己的数据集时出现了以下错误，能帮忙看下么 RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 40 and 48 in dimension 2 at /pytorch/aten/src/THC/generic/THCTensorMath.cu:71

AssertionError

2

## ❓ Questions and Help File "/home/xxd/1 scene graph/Scene-Graph-Benchmark.pytorch-master/maskrcnn_benchmark/data/datasets/visual_genome.py", line 291, in load_image_filenames assert len(fns) == 108073 AssertionError Does anyone encounter this problem?thk

Tensorrt版本

10

您好，请问您用的是tensorrtt5还是6