Quantcast
Channel: TAO Toolkit - NVIDIA Developer Forums
Viewing all articles
Browse latest Browse all 497

Train yolov4 with custom dataset

$
0
0

Please provide complete information as applicable to your setup.

• Hardware Platform (Jetson / GPU) Jetson Orin Nano
• DeepStream Version 6.3
• JetPack Version (valid for Jetson only) 5.1.2
• TensorRT Version
• NVIDIA GPU Driver Version (valid for GPU only)
• Issue Type( questions, new requirements, bugs)

I am training Nvidia models on the dGPU with nvidia tlt stream analytics toolkit docker version 3.21.
I want to train a yolov4 with custom dataset of different shapes. but it gives me this error:

ValueError: Negative dimension size caused by subtracting 7 from 6 for 'conv1/convolution' (op: 'Conv2D') with input shapes: [?,3,6,6], [7,7,3,64].

I tried to convert my dataset to tfrecord and it gives me this error

Message type "DataSource" has no field named "tfrecords_path".
Traceback (most recent call last):

can you please help me with that
Thanks

2 posts - 2 participants

Read full topic


Viewing all articles
Browse latest Browse all 497

Trending Articles