mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
Add activations enum
This commit is contained in:
@@ -16,7 +16,7 @@ class Dense : public ILayer {
|
||||
Dense(
|
||||
int inputSize,
|
||||
int outputSize,
|
||||
std::string activation,
|
||||
Activation activation,
|
||||
cublasHandle_t cublasHandle
|
||||
);
|
||||
~Dense();
|
||||
@@ -37,7 +37,7 @@ class Dense : public ILayer {
|
||||
std::vector<float> weights;
|
||||
std::vector<float> biases;
|
||||
|
||||
std::string activation;
|
||||
Activation activation;
|
||||
|
||||
void initializeWeights();
|
||||
void initializeBiases();
|
||||
|
||||
Reference in New Issue
Block a user