Как / где загружаются файлы контрольных точек

Question

Я использую model_main.py модели tenensflow object_detection. Я хотел бы знать, какой файл .py содержит код, где модель проверяет существующую контрольную точку в 'mode_dir' и как модель узнает, чтобы загрузить / запустить с последней сохраненной контрольной точки.

Я зарегистрировался в model_lib.py, но пока не могу выяснить.

Answer 1

Помимо приведенного выше обсуждения, здесь - это место, где предварительно обученные контрольные точки загружаются в api обнаружения объекта тензорного потока. Как видите, пути контрольных точек передаются переменной train_config.fine_tune_checkpoint, эта переменная является полем под train_config в файле конфигурации. Это не то же самое, что model_dir, который используется только для сохранения самых последних контрольных точек.

Если вы используете ту же командную строку для возобновления обучения, обучение, казалось, автоматически начинается с самой последней контрольной точки в model_dir, и это поддерживается tenorflow Supervisor. См. здесь для справки.