restoring basic TF benchmarking similar to Caffe (run program:tensorflow -> test_cpu/test_cuda)
"run_cmd_out1": "stderr.log",
"run_cmd_out2": "stderr2.log"
}
+ },
+ "test_cpu": {
+ "dataset_tags": [
+ "benchmark",
+ "tensorflow",
+ "dataset",
+ "vcpu"
+ ],
+ "ignore_return_code": "no",
+ "run_deps": {
+ "tensorflow": {
+ "local": "yes",
+ "name": "TensorFlow library",
+ "skip_pruning_by_other_deps": "yes",
+ "sort": 10,
+ "tags": "lib,tensorflow-cpu"
}
},
- "run_vars": {},
+ "run_time": {
+ "run_cmd_main": "$<<CK_ENV_COMPILER_PYTHON_FILE>>$ $#dataset_path#$$#dataset_filename#$ --batch_size=$<<BATCH_SIZE>>$ --num_batches=$<<NUM_BATCHES>>$"
+ }
+ },
+ "test_cuda": {
+ "dataset_tags": [
+ "benchmark",
+ "tensorflow",
+ "dataset",
+ "vcuda"
+ ],
+ "ignore_return_code": "no",
+ "run_deps": {
+ "tensorflow": {
+ "local": "yes",
+ "name": "TensorFlow library",
+ "skip_pruning_by_other_deps": "yes",
+ "sort": 10,
+ "tags": "lib,tensorflow-cuda"
+ }
+ },
+ "run_time": {
+ "run_cmd_main": "$<<CK_ENV_COMPILER_PYTHON_FILE>>$ $#dataset_path#$$#dataset_filename#$ --batch_size=$<<BATCH_SIZE>>$ --num_batches=$<<NUM_BATCHES>>$"
+ }
+ }
+ },
+ "run_vars": {
+ "BATCH_SIZE": 5,
+ "NUM_BATCHES": 5
+ },
"tags": [
"tensorflow-classification",
"ck-ai-image-classification",
