[mlpack-git] [mlpack/mlpack] Inception layer (#757)

Marcus Edel notifications at github.com
Wed Aug 17 17:34:48 EDT 2016


> +    // Forward pass for 3x3 pool path. 
> +    pool3.InputParameter() = input;
> +    pool3.Forward(input, pool3.OutputParameter());
> +    Pad(pool3.OutputParameter(), 1, 1, convPool.InputParameter());
> +    convPool.Forward(pool3.OutputParameter(), convPool.OutputParameter());
> +    biasPool.Forward(convPool.OutputParameter(), biasPool.OutputParameter());
> +    basePool.InputParameter() = biasPool.OutputParameter();
> +    basePool.Forward(convPool.OutputParameter(), basePool.OutputParameter());
> +
> +    // concatenate outputs of all the paths.
> +    output = arma::join_slices( 
> +              arma::join_slices(
> +                arma::join_slices( 
> +                  base1.OutputParameter(), base3.OutputParameter() ), 
> +                  base5.OutputParameter() ), basePool.OutputParameter());
> +

The main class (e.g. CNN) has to initialize the weights somehow, to do that it needs to know the parameter size (weights). So, what the main class basically does is to create a parameter vector, initialize that parameter vector and call the Forward, Backward and Gradient function with the initialized parameters.

-- 
You are receiving this because you are subscribed to this thread.
Reply to this email directly or view it on GitHub:
https://github.com/mlpack/mlpack/pull/757/files/06d923321f246f2c6ead9ad56e99309fe24a6f5c#r75211871
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <https://mailman.cc.gatech.edu/pipermail/mlpack-git/attachments/20160817/ade50001/attachment.html>


More information about the mlpack-git mailing list