diff --git a/LICENSE b/LICENSE
index 059f0fe..c4878e1 100644
--- a/LICENSE
+++ b/LICENSE
@@ -1,6 +1,6 @@
BSD 3-Clause License
-Copyright (c) 2021, Intel Labs
+Copyright (c) 2023, Intel Labs
All rights reserved.
Redistribution and use in source and binary forms, with or without
diff --git a/README.md b/README.md
index 8ef5d7b..13652e9 100644
--- a/README.md
+++ b/README.md
@@ -7,7 +7,7 @@ A library for Bayesian neural network layers and uncertainty estimation in Deep
[![python](https://img.shields.io/badge/python-3.7%2B-blue)](https://github.com/IntelLabs/bayesian-torch)
[![pytorch](https://img.shields.io/badge/pytorch-1.7.0%2B-orange)](https://github.com/IntelLabs/bayesian-torch)
-[![version](https://img.shields.io/badge/release-0.3.0-green)](https://github.com/IntelLabs/bayesian-torch/releases)
+[![version](https://img.shields.io/badge/release-0.4.0-green)](https://github.com/IntelLabs/bayesian-torch/releases)
[![license](https://img.shields.io/badge/license-BSD%203--Clause-blue)](https://github.com/IntelLabs/bayesian-torch/blob/main/LICENSE)
[![Downloads](https://static.pepy.tech/badge/bayesian-torch)](https://pepy.tech/project/bayesian-torch)
diff --git a/bayesian_torch/ao/nn/quantized/modules/quantize_conv_variational.py b/bayesian_torch/ao/nn/quantized/modules/quantize_conv_variational.py
index a8b25dc..7d053c2 100644
--- a/bayesian_torch/ao/nn/quantized/modules/quantize_conv_variational.py
+++ b/bayesian_torch/ao/nn/quantized/modules/quantize_conv_variational.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
@@ -1425,4 +1425,4 @@ def forward(self, input, enable_int8_compute=True, normal_scale=6/255, default_s
if return_kl:
return out, 0 # disable kl divergence computing
- return out
\ No newline at end of file
+ return out
diff --git a/bayesian_torch/ao/nn/quantized/modules/quantize_linear_variational.py b/bayesian_torch/ao/nn/quantized/modules/quantize_linear_variational.py
index e666f9b..abd2269 100644
--- a/bayesian_torch/ao/nn/quantized/modules/quantize_linear_variational.py
+++ b/bayesian_torch/ao/nn/quantized/modules/quantize_linear_variational.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/ao/nn/quantized/modules/quantized_conv_flipout.py b/bayesian_torch/ao/nn/quantized/modules/quantized_conv_flipout.py
index cf771c7..66e203b 100644
--- a/bayesian_torch/ao/nn/quantized/modules/quantized_conv_flipout.py
+++ b/bayesian_torch/ao/nn/quantized/modules/quantized_conv_flipout.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
@@ -1300,4 +1300,4 @@ def forward(self, x, normal_scale=6/255, default_scale=0.1, default_zero_point=1
if return_kl:
return out, 0
- return out
\ No newline at end of file
+ return out
diff --git a/bayesian_torch/ao/nn/quantized/modules/quantized_linear_flipout.py b/bayesian_torch/ao/nn/quantized/modules/quantized_linear_flipout.py
index 289da98..afe1dfa 100644
--- a/bayesian_torch/ao/nn/quantized/modules/quantized_linear_flipout.py
+++ b/bayesian_torch/ao/nn/quantized/modules/quantized_linear_flipout.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/ao/quantization/quantize.py b/bayesian_torch/ao/quantization/quantize.py
index e7f2b64..ff569f4 100644
--- a/bayesian_torch/ao/quantization/quantize.py
+++ b/bayesian_torch/ao/quantization/quantize.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/base_variational_layer.py b/bayesian_torch/layers/base_variational_layer.py
index 8263e82..2606829 100644
--- a/bayesian_torch/layers/base_variational_layer.py
+++ b/bayesian_torch/layers/base_variational_layer.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/flipout_layers/conv_flipout.py b/bayesian_torch/layers/flipout_layers/conv_flipout.py
index 2ad0679..2ea409c 100644
--- a/bayesian_torch/layers/flipout_layers/conv_flipout.py
+++ b/bayesian_torch/layers/flipout_layers/conv_flipout.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/flipout_layers/linear_flipout.py b/bayesian_torch/layers/flipout_layers/linear_flipout.py
index aa6f702..c2a2f8f 100644
--- a/bayesian_torch/layers/flipout_layers/linear_flipout.py
+++ b/bayesian_torch/layers/flipout_layers/linear_flipout.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/flipout_layers/quantized_conv_flipout.py b/bayesian_torch/layers/flipout_layers/quantized_conv_flipout.py
index 4be011a..6246a8d 100644
--- a/bayesian_torch/layers/flipout_layers/quantized_conv_flipout.py
+++ b/bayesian_torch/layers/flipout_layers/quantized_conv_flipout.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
@@ -1348,4 +1348,4 @@ def forward(self, x, normal_scale=6/255, default_scale=0.1, default_zero_point=1
if return_kl:
return out, 0
- return out
\ No newline at end of file
+ return out
diff --git a/bayesian_torch/layers/flipout_layers/quantized_linear_flipout.py b/bayesian_torch/layers/flipout_layers/quantized_linear_flipout.py
index 3cce873..2edd57e 100644
--- a/bayesian_torch/layers/flipout_layers/quantized_linear_flipout.py
+++ b/bayesian_torch/layers/flipout_layers/quantized_linear_flipout.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/flipout_layers/rnn_flipout.py b/bayesian_torch/layers/flipout_layers/rnn_flipout.py
index 5977740..1c47135 100644
--- a/bayesian_torch/layers/flipout_layers/rnn_flipout.py
+++ b/bayesian_torch/layers/flipout_layers/rnn_flipout.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/variational_layers/conv_variational.py b/bayesian_torch/layers/variational_layers/conv_variational.py
index 403651a..71249dd 100644
--- a/bayesian_torch/layers/variational_layers/conv_variational.py
+++ b/bayesian_torch/layers/variational_layers/conv_variational.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/variational_layers/linear_variational.py b/bayesian_torch/layers/variational_layers/linear_variational.py
index 4cb1adb..5cc5d02 100644
--- a/bayesian_torch/layers/variational_layers/linear_variational.py
+++ b/bayesian_torch/layers/variational_layers/linear_variational.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/variational_layers/quantize_conv_variational.py b/bayesian_torch/layers/variational_layers/quantize_conv_variational.py
index 31ed9e7..e3ca3c1 100644
--- a/bayesian_torch/layers/variational_layers/quantize_conv_variational.py
+++ b/bayesian_torch/layers/variational_layers/quantize_conv_variational.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
@@ -1489,4 +1489,4 @@ def forward(self, input, enable_int8_compute=True, normal_scale=6/255, default_s
if return_kl:
return out, 0 # disable kl divergence computing
- return out
\ No newline at end of file
+ return out
diff --git a/bayesian_torch/layers/variational_layers/quantize_linear_variational.py b/bayesian_torch/layers/variational_layers/quantize_linear_variational.py
index a12a569..7d48cc3 100644
--- a/bayesian_torch/layers/variational_layers/quantize_linear_variational.py
+++ b/bayesian_torch/layers/variational_layers/quantize_linear_variational.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/layers/variational_layers/rnn_variational.py b/bayesian_torch/layers/variational_layers/rnn_variational.py
index 39f4a2d..e7ca030 100644
--- a/bayesian_torch/layers/variational_layers/rnn_variational.py
+++ b/bayesian_torch/layers/variational_layers/rnn_variational.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/models/bnn_to_qbnn.py b/bayesian_torch/models/bnn_to_qbnn.py
index 85953cf..d18a9a4 100644
--- a/bayesian_torch/models/bnn_to_qbnn.py
+++ b/bayesian_torch/models/bnn_to_qbnn.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
@@ -256,4 +256,4 @@ def forward(self, x):
print(m)
if input.dtype!=torch.quint8:
input = torch.quantize_per_tensor(input, 0.1, 128, torch.quint8)
- print(m(input))
\ No newline at end of file
+ print(m(input))
diff --git a/bayesian_torch/models/dnn_to_bnn.py b/bayesian_torch/models/dnn_to_bnn.py
index 92e18b4..8077581 100644
--- a/bayesian_torch/models/dnn_to_bnn.py
+++ b/bayesian_torch/models/dnn_to_bnn.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/scripts/quantize_bayesian_cifar.sh b/bayesian_torch/scripts/quantize_bayesian_cifar.sh
new file mode 100644
index 0000000..39854a2
--- /dev/null
+++ b/bayesian_torch/scripts/quantize_bayesian_cifar.sh
@@ -0,0 +1,9 @@
+#!/bin/bash
+
+model=resnet20
+mode='ptq'
+batch_size=1000
+num_monte_carlo=20
+checkpoint_file=./checkpoint/bayesian/bayesian_resnet20_cifar.pth
+
+python examples/main_bayesian_cifar_dnn2bnn.py --arch=$model --mode=$mode --batch-size=$batch_size --num_monte_carlo=$num_monte_carlo --model-checkpoint=$checkpoint_file
diff --git a/bayesian_torch/utils/avuc_loss.py b/bayesian_torch/utils/avuc_loss.py
index 5a4dc75..322cb97 100644
--- a/bayesian_torch/utils/avuc_loss.py
+++ b/bayesian_torch/utils/avuc_loss.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Corporation
+# Copyright (C) 2023 Intel Corporation
#
# BSD-3-Clause License
#
diff --git a/bayesian_torch/utils/util.py b/bayesian_torch/utils/util.py
index 7418679..df60823 100644
--- a/bayesian_torch/utils/util.py
+++ b/bayesian_torch/utils/util.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2021 Intel Labs
+# Copyright (C) 2023 Intel Labs
#
# BSD-3-Clause License
#
diff --git a/setup.py b/setup.py
index 6629022..202055e 100644
--- a/setup.py
+++ b/setup.py
@@ -11,7 +11,7 @@
setup(
name = "bayesian-torch",
packages = find_packages(),
- version = "0.3.0",
+ version = "0.4.0",
description = "A library for Bayesian neural network layers and uncertainty estimation in Deep Learning",
author = "Intel Labs",
author_email = "ranganath.krishnan@intel.com",