Implement simple model validation

This commit is contained in:
2024-04-22 20:57:40 +02:00
parent f17debc244
commit c2acad151b
3 changed files with 24 additions and 0 deletions

View File

@@ -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