Skip to content

Commit

Permalink
Code for unprune yolo
Browse files Browse the repository at this point in the history
osw282 committed Jan 29, 2025

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent 4b31a0e commit 4f4cedb
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion jetson/power_logging/model/benchmark_pruned_model.py
Original file line number Diff line number Diff line change
@@ -180,7 +180,7 @@ def benchmark(args: argparse.Namespace) -> None:
# amount=0.5,
# )
# Thus should load the pruned yolo model
model = torch.load("pruned_30_yolov5su.pt")
model = torch.load("yolov5su.pt")
model.eval().to(DEVICE)

dtype = torch.float32
4 changes: 2 additions & 2 deletions jetson/power_logging/run_experiment.sh
Original file line number Diff line number Diff line change
@@ -23,7 +23,7 @@ sleep 120
# NOTE : fcn_resnet50 is a object detection model and does not work with TorchTensorRT library
# TODO: Revisit fcn_resnet50 once this issue is addressed: https://github.com/pytorch/TensorRT/issues/3295
#models=("alexnet" "vgg11" "vgg13" "vgg16" "vgg19" "mobilenet_v2" "mobilenet_v3_small" "mobilenet_v3_large" "resnet18" "resnet34" "resnet50" "resnet101" "resnet152" "lenet" "resnext50_32x4d" "resnext101_32x8d" "resnext101_64x4d" "convnext_tiny" "convnext_small" "convnext_base")
models=("lenet")
models=("yolov5s")
# Number of inference cycles
RUNS=30000

@@ -36,7 +36,7 @@ do
if [ "$model" == "lenet" ]; then
INPUT_SHAPE='--input-shape 1 1 32 32'
else
INPUT_SHAPE='--input-shape 1 3 224 224'
INPUT_SHAPE='--input-shape 1 3 640 640'
fi

# Run the measure_inference_power.py script

0 comments on commit 4f4cedb

Please sign in to comment.