使用训练好的模型作为其他图像分类的预训练要怎么修改模型?
Created by: yeyupiaoling
官方在这里提供了一些训练好的模型:https://github.com/PaddlePaddle/models/tree/develop/fluid/image_classification#supported-models-and-performances 我使用这些模型作为其他类别和图片大小的与训练模型,我参考这里这个代码: https://github.com/PaddlePaddle/models/blob/8f61de12fdd35ce9aa5073263724fad6b73ff8bf/fluid/image_classification/train.py#L147-L152
按照之前的方式使用了ResNet50,把最好的全连接层的文件删除,在运行的时候结果报错:
*** Aborted at 1533809394 (unix time) try "date -d @1533809394" if you are using GNU date ***
PC: @ 0x0 (unknown)
*** SIGABRT (@0x4acb) received by PID 19147 (TID 0x7f0eee645700) from PID 19147; stack trace: ***
@ 0x7f0eee22c390 (unknown)
@ 0x7f0eede86428 gsignal
@ 0x7f0eede8802a abort
@ 0x7f0eedec87ea (unknown)
@ 0x7f0eeded137a (unknown)
@ 0x7f0eeded553c cfree
@ 0x7f0ec076b472 paddle::framework::details::ThreadedSSAGraphExecutor::Run()
@ 0x7f0ec076fca5 paddle::framework::details::ScopeBufferedSSAGraphExecutor::Run()
@ 0x7f0ebfb7285d paddle::framework::ParallelExecutor::Run()
@ 0x7f0ebfaf41d6 _ZZN8pybind1112cpp_function10initializeIZNS0_C1IvN6paddle9framework16ParallelExecutorEIRKSt6vectorISsSaISsEERKSsEINS_4nameENS_9is_methodENS_7siblingEEEEMT0_FT_DpT1_EDpRKT2_EUlPS5_SA_SC_E_vISQ_SA_SC_EISD_SE_SF_EEEvOSH_PFSG_SJ_ESP_ENUlRNS_6detail13function_callEE1_4_FUNESX_
@ 0x7f0ebfacc604 pybind11::cpp_function::dispatcher()
@ 0x4c37ed PyEval_EvalFrameEx
@ 0x4b9ab6 PyEval_EvalCodeEx
@ 0x4c16e7 PyEval_EvalFrameEx
@ 0x4b9ab6 PyEval_EvalCodeEx
@ 0x4c1e6f PyEval_EvalFrameEx
@ 0x4b9ab6 PyEval_EvalCodeEx
@ 0x4eb30f (unknown)
@ 0x4e5422 PyRun_FileExFlags
@ 0x4e3cd6 PyRun_SimpleFileExFlags
@ 0x493ae2 Py_Main
@ 0x7f0eede71830 __libc_start_main
@ 0x4933e9 _start
@ 0x0 (unknown)
训练好的模型要怎样修改才能作为自己分类任务的预训练模型?