인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

인프런 커뮤니티 질문&답변

genome87님의 프로필 이미지
genome87

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

yolov3 pytorch 관련 질문 custom dataset 중

작성

·

345

0

강의에는 없는 듯 하나 pytorch yolov3 로 custom data를 학습하려는데 

자꾸 에러가 납니다. 

 File "E:\source\pysource\PyTorch-YOLOv3-master\PyTorch-YOLOv3-master\models.py", line 263, in forward

  yolo_outputs = to_cpu(torch.cat(yolo_outputs, 1)) 

RuntimeError: There were no tensor arguments to this function (e.g., you passed an empty list of Tensors), but no fallback function is registered for schema aten::_cat.  This usually means that this function requires a non-empty list of Tensors.  Available functions are [CPUTensorId, QuantizedCPUTensorId, VariableTensorId]

print로 해본 결과 입력에 들어가는 이미지 값들은 값이 나오는데 왜 tensor argument가 empty로 전달되었는지를 모르겠습니다. 

File "E:/source/pysource/PyTorch-YOLOv3-master/PyTorch-YOLOv3-master/train.py", line 111, in <module>

 loss, outputs = model(imgs, targets)

답변 1

0

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까,

안타깝게도 제가 pytorch는 능력이 되지 않아 도움이 못될 것 같습니다.

감사합니다.

genome87님의 프로필 이미지
genome87

작성한 질문수

질문하기