ArcGIS Pro crash after running Train Deep Learning Model

818
0
05-13-2020 11:54 AM
JoelLangdon
New Contributor

Hey,

My ArcGIS Pro and python shell both successfully output the training model, but crash on completion. This is a problem for scripting, as I can't run the next step (classify objects using deep learning) within the same script. When it crashes, no .dmp is created and there is no error message.

##SSD SETTINGS
max_epochs = 25
model_type = "SSD"
batch_size = 5
arg = ""
learning_rate = 0.001
backbone_model = "RESNET34"
validation_percent = 5
stop_training = "CONTINUE_TRAINING"

try:
    TrainDeepLearningModel(training_data, modeldir, max_epochs, model_type,
        batch_size, arg, learning_rate, backbone_model, "",
        validation_percent, stop_training)
except Exception:
    e = sys.exc_info()[1]
    print(e.args[0])

This is the block of code in question. I also don't get an error message from except. I'm using a GTX960 and my CPU is a Ryzen 5 2600x. I have the same problem using GUI as well.

Anyone have any advice or insight into what's happening here?

Thanks

0 Kudos
0 Replies