Open whwususu opened 3 days ago
您好,可以标注自己的数据集进行训练,可以参考这里面的版面区域定位文档https://github.com/PaddlePaddle/PaddleX/blob/release/3.0-beta1/docs/module_usage/tutorials/ocr_modules/layout_detection.md和 标注文档 https://github.com/PaddlePaddle/PaddleX/blob/release/3.0-beta1/docs/data_annotations/cv_modules/object_detection.md。
MemoryError:
C++ Traceback (most recent call last):
0 paddle::pybind::eager_api_conv2d(_object, _object, _object)
1 conv2d_ad_func(paddle::Tensor const&, paddle::Tensor const&, std::vector<int, std::allocator >, std::vector<int, std::allocator >, std::string, std::vector<int, std::allocator >, int, std::string)
2 paddle::experimental::conv2d(paddle::Tensor const&, paddle::Tensor const&, std::vector<int, std::allocator > const&, std::vector<int, std::allocator > const&, std::string const&, std::vector<int, std::allocator > const&, int, std::string const&)
3 void phi::ConvCudnnKernel<float, phi::GPUContext>(phi::GPUContext const&, phi::DenseTensor const&, phi::DenseTensor const&, std::vector<int, std::allocator > const&, std::vector<int, std::allocator > const&, std::string const&, std::vector<int, std::allocator > const&, int, std::string const&, phi::DenseTensor)
4 void phi::ConvCudnnKernelImplV7<float, phi::GPUContext>(phi::DenseTensor const, phi::DenseTensor const, phi::GPUContext const&, std::vector<int, std::allocator > const&, std::vector<int, std::allocator > const&, std::vector<int, std::allocator > const&, phi::backends::gpu::DataLayout, phi::backends::gpu::DataLayout, bool, bool, int, phi::DenseTensor)
5 phi::DnnWorkspaceHandle::ReallocWorkspace(unsigned long)
6 paddle::memory::allocation::Allocator::Allocate(unsigned long)
7 paddle::memory::allocation::StatAllocator::AllocateImpl(unsigned long)
8 paddle::memory::allocation::Allocator::Allocate(unsigned long)
9 paddle::memory::allocation::Allocator::Allocate(unsigned long)
10 paddle::memory::allocation::Allocator::Allocate(unsigned long)
11 paddle::memory::allocation::Allocator::Allocate(unsigned long)
12 paddle::memory::allocation::CUDAAllocator::AllocateImpl(unsigned long)
13 std::string phi::enforce::GetCompleteTraceBackString
Error Message Summary:
麻烦您看下,我换成了3090显卡,还是不行,有不有参数调下参数,让他训练下去,谢谢!
问题确认 Search before asking
请提出你的问题 Please ask your question
请问下,版面分析模型的训练方法我看了半天不得其解,都是提供了各种数据集进行训练,而不是自定义训练集。那么有不有例如通过打标的方式,生成自己的训练集,然后进行训练的。 这个是我看的教程。https://github.com/PaddlePaddle/PaddleOCR/blob/release/2.6/ppstructure/layout/README_ch.md 谢谢!