mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
Implement simple model validation
This commit is contained in:
@@ -109,6 +109,9 @@ int main(int argc, const char* const argv[]) {
|
||||
const int outputSize = 1000;
|
||||
|
||||
CUDANet::Model *model = createModel(inputSize, inputChannels, outputSize);
|
||||
|
||||
model->validate();
|
||||
|
||||
model->loadWeights(modelWeightsPath);
|
||||
|
||||
// Read and normalize the image
|
||||
|
||||
Reference in New Issue
Block a user