File size: 780 Bytes
22ef5f8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | {
"name": "baseline",
"source_checkpoint": "outputs/train/baseline/weights/best.pt",
"engine": "outputs/tensorrt_fp16/baseline/model.engine",
"input_shape": [
1,
3,
640,
640
],
"output_shape_before_export": [
1,
10,
8400
],
"class_count": 6,
"class_names": [
"open",
"short",
"mousebite",
"spur",
"copper",
"pin-hole"
],
"precision": "fp16",
"dynamic": false,
"batch_size": 1,
"parameters": 3012018,
"counted_parameters": 3012018,
"macs": 4073266400,
"gmacs": 4.0732664,
"flops_estimate": 8146532800,
"gflops_estimate": 8.1465328,
"gpu_name": "Tesla T4",
"torch_version": "2.10.0+cu128",
"cuda_version": "12.8",
"tensorrt_version": "10.16.1.11",
"ultralytics_version": "8.4.115"
} |