Эквивалент слоя DummyData в Caffe2?

Question

Я пытаюсь преобразовать довольно сложную сверточную нейронную сеть из Caffe в Caffe2 по соображениям производительности. Я знаю учебник по преобразованию эквивалентных слоев, однако, мои слои DummyData в Caffe прерывают преобразование. Хотя я знаю, что мне, скорее всего, придется расширять учебный сценарий, чтобы учесть это, я надеялся, что кто-то может указать мне правильное направление для создания эквивалента этого типа слоя в Caffe2.

Мне также нужно конвертировать пару разных слоев Python, так что довольно неизбежно, что мне нужно будет изменить учебник в репозитории Caffe2 Github; Я просто не уверен, как подойти к этому, и подумал, что было бы очень полезно увидеть, как работает слой DummyData. К сожалению, у меня нет доступа к написанному мною коду, так как он находится на моем компьютере в работе, однако все, что я до сих пор делал, это добавлял новую функцию с аргументами, аналогичными тем, которые были даны всем другим слоям, и создавал новый оператор. Я не уверен, что это правильный подход (поскольку может существовать в значительной степени эквивалентный существующий оператор, который я пропустил).

Answer 1

Я не проверял это, но это может сработать: добавить UniformFillOp (для создания фиктивных данных), а затем передать его вывод в ImageInput (для минибата и увеличения данных).

Список операторов, которые вы можете использовать для создания фиктивных данных: https://github.com/caffe2/caffe2/blob/master/caffe2/operators/filler_op.cc#L17-L25