-
Notifications
You must be signed in to change notification settings - Fork 43
Open
Description
Due Date
To be completed by: YYYY-MM-DD
Description
I tried the example at https://deeplearning4j.org/model-zoo, i.e. ZooModel zooModel = new AlexNet(..); MultiLayerConfiguration net = zooModel.conf();
Sadly, the conf() method is not part of the ZooModel class. I use 1.0.0-alpha.
I want to use the zoo model for regression somehow, and the idea was to change the output layer with the configuration accordingly. Now there is no general, thus configurable possibility left for this.
Best would be a 'regression switch'/flag for the zoo models, e.g. 'setRegression(true)' in class ZooModel.
Assignees
Please ensure you have assigned at least one person to this issue. Include any authors and reviewers required.
Metadata
Metadata
Assignees
Labels
No labels