LaneNet:使用其他数据集出现错误
Created by: iceriver97
我使用官方的LaneNet车道线分割项目,但更换了自己的数据集,格式都按照官方项目数据集的格式整理好了,但报错:
#Device count: 1
batch_size_per_dev: 2
W0924 18:20:06.613276 12848 device_context.cc:252] Please NOTE: device: 0, CUDA Capability: 75, Driver API Version: 10.2, Runtime API Version: 10.0
W0924 18:20:06.621250 12848 device_context.cc:260] device: 0, cuDNN Version: 7.6.
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/score_origin/weights doesn't exist
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/deconv_1/weights doesn't exist
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/score_1/weights doesn't exist
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/deconv_2/weights doesn't exist
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/score_2/weights doesn't exist
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/deconv_final/weights doesn't exist
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/score_final/weights doesn't exist
[SKIP] Pretrained weight ./pretrained_models/VGG16_pretrained/weights doesn't exist
2020-09-24 18:20:08,732-WARNING: ./pretrained_models/VGG16_pretrained.pdparams not found, try to load model file saved with [ save_params, save_persistables, save_vars ]
There are 13/21 varaibles in ./pretrained_models/VGG16_pretrained are loaded.
Use multi-thread reader
W0924 18:20:08.925722 12848 build_strategy.cc:170] fusion_group is not enabled for Windows/MacOS now, and only effective when running with CUDA GPU.
epoch=1 step=10 lr=0.00050 loss=242.3553 seg_loss=10.7224 emb_loss=231.6328 accuracy=0.383 fp=0.9665 fn=0.617 step/sec=5.359 | ETA 11:22:36
W0924 18:20:11.018095 12848 operator.cc:187] cast raises an exception class thrust::system::system_error, parallel_for failed: invalid configuration argument
F0924 18:20:11.019109 12848 exception_holder.h:37] std::exception caught, parallel_for failed: invalid configuration argument
*** Check failure stack trace: ***
这是什么原因呢?我在AIStudio上运行也是同样的错误