Merge pull request #23 from IntelLabs/quantization

Add support for Bayesian neural network quantization
IntelLabs · Jun 27, 2023 · b0fc3a1 · b0fc3a1
2 parents aa8e198 + d9a26a5
commit b0fc3a1
Show file tree

Hide file tree

Showing 30 changed files with 8,524 additions and 24 deletions.
diff --git a/bayesian_torch/__init__.py b/bayesian_torch/__init__.py
@@ -0,0 +1 @@
+from bayesian_torch import quantization as quantization
diff --git a/bayesian_torch/ao/__init__.py b/bayesian_torch/ao/__init__.py
diff --git a/bayesian_torch/ao/nn/__init__.py b/bayesian_torch/ao/nn/__init__.py
diff --git a/bayesian_torch/ao/nn/quantized/__init__.py b/bayesian_torch/ao/nn/quantized/__init__.py
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from bayesian_torch import quantization as quantization