updates for v0.4.0 release preparation

ranganathkrishnan · ranganathkrishnan · commit 93cf0d3d43d3 · 2023-07-27T00:17:34.000-07:00
Signed-off-by: Ranganath Krishnan &lt;ranganath.krishnan@intel.com&gt;
diff --git 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
@@ -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)
 <h4 align="center">
diff --git 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
+        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
@@ -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
@@ -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
+        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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
+        return out
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
+        return out
diff --git 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
@@ -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
@@ -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))
+    print(m(input))
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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",

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-# Copyright (C) 2021 Intel Labs`
	`1`	`+# Copyright (C) 2023 Intel Labs`
`2`	`2`	`#`
`3`	`3`	`# BSD-3-Clause License`
`4`	`4`	`#`
`@@ -1425,4 +1425,4 @@ def forward(self, input, enable_int8_compute=True, normal_scale=6/255, default_s`
`1425`	`1425`	`if return_kl:`
`1426`	`1426`	`return out, 0 # disable kl divergence computing`
`1427`	`1427`
`1428`		`- return out`
	`1428`	`+ return out`