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