Skip to content

Instantly share code, notes, and snippets.

@wvdschel
Created December 31, 2023 09:14
Show Gist options
  • Save wvdschel/368cd512e71da325cea056bd810d01a1 to your computer and use it in GitHub Desktop.
Save wvdschel/368cd512e71da325cea056bd810d01a1 to your computer and use it in GitHub Desktop.
pytorch laptop benchmarks
for model in llama resnet50 resnet152 vgg16 hf_gpt2 hf_gpt2_large hf_bert yolov3; do python run.py -d cuda -t eval $model; python run.py -d cuda -t train $model; done 2>&1 | tee benchmark.log
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment