EarlyStoppingTrainer

DL4J
1

Hi everyone,

Is there any way to return a MultiLayerNetwork from using the EarlyStoppingTrainer? From what I understand the trainer is only able to return a Model object, and I don’t see any way to convert the model to the network object for use in evaluating the testing data…

Best regards!

2

Depending on the network configuration you use the trainer will return to you either a CompuationGraph or a MultiLayerNetwork, as you know exactly which one is will be, you can just cast that to the appropriate class in order to access any specific methods.