Skip to content

Commit 514a513

Browse files
author
Seth Weidman
committed
Add comment on no GPUs in static quantization tutorial
1 parent 32949f5 commit 514a513

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

advanced_source/static_quantization_tutorial.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,8 @@
88
99
This tutorial shows how to do post-training static quantization, as well as illustrating
1010
two more advanced techniques - per-channel quantization and quantization-aware training -
11-
to further improve the model's accuracy.
11+
to further improve the model's accuracy. Note that quantization is currently only supported
12+
for CPUs, so we will not be utilizing GPUs / CUDA in this tutorial.
1213
1314
By the end of this tutorial, you will see how quantization in PyTorch can result in
1415
significant decreases in model size while increasing speed. Furthermore, you'll see how

0 commit comments

Comments
 (0)