Save
I've created the optimizer class
This commit is contained in:
@@ -37,7 +37,7 @@
|
||||
// INCLUDE DESCRIPTION
|
||||
//---------------------------------------------------------------------------------------------------------------------------
|
||||
#include <config/types.hpp> // panic::uint_t, panic::int_t, and panic::real_t
|
||||
#include <neural_network/layer/layer.hpp> // for base layer struct
|
||||
#include <neural_network/layer/trainable_layer.hpp> // for base trainable_layer struct
|
||||
#include <tensor/vector.hpp>
|
||||
#include <tensor/matrix.hpp>
|
||||
|
||||
@@ -56,7 +56,7 @@ namespace panic{
|
||||
*
|
||||
* The struct is used in PANIC nural_network library.
|
||||
*/
|
||||
struct layer_dense : public layer{
|
||||
struct layer_dense : trainable_layer{
|
||||
|
||||
/**
|
||||
* @brief Emphty matrix to store input data
|
||||
|
||||
Reference in New Issue
Block a user