Unable to run train_mrcnn_model.ipynb #20
Description
When running train_mrcnn_model.ipynb with tensorflow gpu I got the following erros.
import warnings
with warnings.catch_warnings():
warnings.simplefilter("ignore")
# Train network heads
print('Training network heads')
model.train(dataset, test_dataset,
learning_rate=config.LEARNING_RATE,
epochs=15,
layers='heads')
# Finetune layers from ResNet stage 4 and up
print('Tuning stage 4 and up')
model.train(dataset, test_dataset,
learning_rate=config.LEARNING_RATE / 5,
epochs=50,
layers='4+')
print('Tuning all layers')
model.train(dataset, test_dataset,
learning_rate=config.LEARNING_RATE / 50,
epochs=100,
layers='all')