Println("Training the network with some data.") Println("Inferring network output for some given inputs.") setInputType(nvolutional(15, 80, 3, format)) Val configuration = NeuralNetConfiguration.Builder()ĬonvolutionLayer(3, 10, intArrayOf(7, 3), convolutionActivation, format),ĬonvolutionLayer(10, 20, intArrayOf(3, 3), convolutionActivation, format), Val convolutionActivation = Activation.RELU Here’s a small self-contained example (written in Kotlin) to illustrate the problem I’m currently facing: import .conf.CNN2DFormat However, I’ve run into some issue where training the network causes my application to crash. I’m trying to train a convolutional neural network (using deeplearning4j-core and nd4j-native-platform both in version 1.0.0-beta7) where the input data is formatted in CNN2DFormat.NHWC.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |