MagmaDNN  1.0
c++NeuralNetworkFramework
optimizers.h
Go to the documentation of this file.
1 
9 #pragma once
10 
11 #include "optimizer/optimizer.h"
13 
14 namespace magmadnn {
15 namespace optimizer {
16 
17 enum optimizer_t {
18  SGD,
19  ADAM,
20 };
21 
22 enum loss_t {
23  CROSS_ENTROPY,
24  MSE
25 };
26 
27 } // namespace optimizer
28 } // namespace magmadnn
Definition: addop.cpp:11