Skip to content
Snippets Groups Projects
Commit eea13361 authored by Axel Farrugia's avatar Axel Farrugia
Browse files

[Fix] Not enough samples in dataset when using mock_db option

parent 14adfae5
No related branches found
No related tags found
1 merge request!64Fix parameter redifinition
......@@ -196,7 +196,7 @@ backend = "cuda" if USE_CUDA else "cpu"
aidge_tensors = []
labels = []
if args.mock_db:
for i in range(NB_TEST):
for i in range(max(NB_TEST, NB_CALIB)):
aidge_tensor = aidge_core.Tensor(dims=(1, 3, 224, 224))
aidge_tensor.set_backend(backend)
aidge_tensor.set_datatype(aidge_core.dtype.float32)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment