Ошибка CHECK при записи пользовательского слоя потерь в Caffe

Я получаю следующую ошибку при запуске тестового примера для пользовательской функции потерь в caffe. Эта функция потерь использует слой Reshape на bottom большом двоичном объекте для этого слоя (это упоминается в методе LayerSetUp() пользовательской функции потерь.

Ошибка:

F0704 01:49:54.075613 16977 blob.cpp:145] Check failed: diff_ 
*** Check failure stack trace: ***
    @     0x7f627a1965cd  google::LogMessage::Fail()
    @     0x7f627a198433  google::LogMessage::SendToLog()
    @     0x7f627a19615b  google::LogMessage::Flush()
    @     0x7f627a198e1e  google::LogMessageFatal::~LogMessageFatal()
    @     0x7f62754dd96b  caffe::Blob<>::mutable_cpu_diff()
    @           0x500e8a  caffe::CustomLossLayerTest_TestRead_Test<>::TestBody()
    @           0x940693  testing::internal::HandleExceptionsInMethodIfSupported<>()
    @           0x939caa  testing::Test::Run()
    @           0x939df8  testing::TestInfo::Run()
    @           0x939ed5  testing::TestCase::Run()
    @           0x93b1af  testing::internal::UnitTestImpl::RunAllTests()
    @           0x93b4d3  testing::UnitTest::Run()
    @           0x46f5fd  main
    @     0x7f6274811830  __libc_start_main
    @           0x477229  _start
    @              (nil)  (unknown)
Makefile:532: recipe for target 'runtest' failed
make: *** [runtest] Aborted (core dumped)

Что может быть возможной причиной?


person harman    schedule 03.07.2017    source источник
comment
@Shai Я забыл настроить (вызвать метод SetUp) одного из слоев Reshape, который я использовал внутри, и в этом была проблема. Сейчас такой ошибки нет.   -  person harman    schedule 05.07.2017
comment
@Shai У меня было еще одно небольшое сомнение: я добавил слой softmax в свой пользовательский слой, который принимает входную часть bottom[0] blob. Я выделил память этому BLOB-объекту с помощью оператора new. Должен ли я беспокоиться о freeing памяти и где именно я должен удалить это выделение памяти?   -  person harman    schedule 06.07.2017
comment
интересный вопрос. Но я понятия не имею, как выглядит ваш код... Мне кажется, это новый вопрос.   -  person Shai    schedule 06.07.2017
comment
Вот ссылка на вопрос: вопрос   -  person harman    schedule 06.07.2017