Модель MaskRcnn не тренируется с «последней»

Question

Я пытаюсь обучить модель maskRcnn на изображениях меланомы. Изначально я обучаю только главам сетей, а затем всей сети. Однако код обучает только заголовки сети.

model.train(dataset_train,dataset_val,
            learning_rate=config.LEARNING_RATE,
            epochs=30,
            layers='heads')

следующий код:

model.train(dataset_train,dataset_val,
            learning_rate=config.LEARNING_RATE,
            epochs=30,
            layers='all')

Не обучает модель и не выдает ошибку. Исполнение просто заканчивается. Весь сценарий можно найти здесь