Please use a new result folder and retry.
More , could you change pred_num_channels as below?
pred_num_channels: 0
Could you share your latest training spec file?
Could you share the full command and full log?
Hi,
Could you try to train with SSD network to check if there is the same error?
Please refer to the SSD training spec file inside jupyter notebook.
I tried with kitti dataset and my own dataset but got error
custom dataset (original image size 500x500):
ssd (copy).ipynb (90.1 KB)
with kiiti dataset:
ssd.ipynb (191.2 KB)
For ssd or dssd, please
-
use the smallest input-size to avoid gpu memory issue as much as possible
-
also change tfrecord format to sequence format.
data_sources: {
tfrecords_path: “/workspace/tao-experiments/Data/kitti_data/tfrecords/kitti_train*”
}to
data_sources: { label_directory_path: "/workspace/tao-experiments/Data/kitti_data/training/label" image_directory_path: "/workspace/tao-experiments/Data/kitti_data/training/image" }
So, issue is gone, right?
Yes, no error occurred
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.