MagmaDNN
1.0
c++NeuralNetworkFramework
optimizers.h
Go to the documentation of this file.
1
9
#pragma once
10
11
#include "
optimizer/optimizer.h
"
12
#include "
optimizer/gradientdescent/gradientdescent.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
magmadnn
Definition:
addop.cpp:11
optimizer.h
gradientdescent.h
include
optimizer
optimizers.h
Generated by
1.8.13