mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 17:54:27 +00:00
Add Kernels namespace
This commit is contained in:
@@ -6,6 +6,17 @@
|
||||
|
||||
namespace Layers {
|
||||
|
||||
enum Activation {
|
||||
SIGMOID,
|
||||
RELU,
|
||||
NONE
|
||||
};
|
||||
|
||||
enum Padding {
|
||||
SAME,
|
||||
VALID
|
||||
};
|
||||
|
||||
class ILayer {
|
||||
public:
|
||||
virtual ~ILayer() {}
|
||||
@@ -29,7 +40,7 @@ class ILayer {
|
||||
std::vector<float> weights;
|
||||
std::vector<float> biases;
|
||||
|
||||
Activation activation;
|
||||
Layers::Activation activation;
|
||||
};
|
||||
|
||||
} // namespace Layers
|
||||
|
||||
Reference in New Issue
Block a user