Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix adding module layers

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-29 21:02:26 +02:00
parent 4159454842
commit 8ac2da004c
4 changed files with 31 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
examples/inception_v3/inception_v3.cpp
View File

@@ -1,16 +1,6 @@
#include <cudanet.cuh>
#include <iostream>
int main(int argc, const char *const argv[]) {
if (argc != 3) {
std::cerr << "Usage: " << argv[0] << "<model_weights_path> <image_path>"
<< std::endl;
return 1; // Return error code indicating incorrect usage
}
std::cout << "Loading model..." << std::endl;
}
class BasicConv2d : public CUDANet::Module {
public:
BasicConv2d(
@@ -702,7 +692,7 @@ class InceptionE : public CUDANet::Module {
}
shape2d getOutputDims() {
branch3x3_2a->getOutputDims();
return branch3x3_2a->getOutputDims();
}
int getOutputChannels() {
@@ -823,7 +813,7 @@ class InceptionV3 : public CUDANet::Model {
Mixed_7c->getOutputSize(), 1000,
CUDANet::Layers::ActivationType::SOFTMAX
);
addLayer("", fc);
addLayer("fc", fc);
}
float* predict(const float* input) {
@@ -905,4 +895,20 @@ class InceptionV3 : public CUDANet::Model {
InceptionE *Mixed_7c;
CUDANet::Layers::Dense *fc;
};
};
int main(int argc, const char *const argv[]) {
InceptionV3 *inception_v3 = new InceptionV3({299, 299}, 3, 1000);
inception_v3->printSummary();
if (argc != 3) {
std::cerr << "Usage: " << argv[0] << "<model_weights_path> <image_path>"
<< std::endl;
return 1; // Return error code indicating incorrect usage
}
std::cout << "Loading model..." << std::endl;
}