caffe забывает установить переменную на этапе тестирования

Question

В caffe, если я создаю свою собственную переменную, связанную со слоем, и изменяю эту переменную во время обучающих итераций, фаза тестирования не знает об изменениях этой переменной, она, кажется, инициализирует ее для себя. этапы тренировочного тестирования делятся весами, но не более того. Как мне разделить другие переменные между этапами обучения и тестирования. Благодарю.

например: скажу, что я объявляю в inner_product_layer.hpp,

int dog = 5;

и в Backward_gpu() я изменяю,

dog = 6;

фаза тестирования думает, что собака =5. Я хочу, чтобы на этапе тестирования было известно, что собаке сейчас 6.

Answer 1

Исполняемый файл для фазы обучения и тестирования отличается, поэтому, если вы измените переменную внутри кода во время обучения, она не будет отражена во время теста. Вы можете сбросить переменную в выходной файл, когда обучение закончится, и прочитать файл в качестве аргумента во время теста при инициализации слоя в функции LayerSetUp. Имя файла может быть добавлено в качестве параметра слоя, где вы можете проверить, находитесь ли вы в фазе тестирования и установить свои параметры из файла, который вы сохранили.