Merge pull request #21 from Vivswan/master

v1.0.1 (Patches for PyTorch 2.0.0)
Vivswan · Mar 27, 2023 · cbbd394 · cbbd394
2 parents 8ed52ea + 68aa611
commit cbbd394
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 6 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,7 @@
 # Changelog
 
 ## 1.0.0
+* Public release
 
-* Public release
+## 1.0.1 (Patches for Pytorch 2.0.0)
+* added `grad.setter` to `PseudoParameterModule` class
diff --git a/README.md b/README.md
@@ -21,7 +21,7 @@ pip install analogvnn
 
 ![3 Layered Linear Photonic Analog Neural Network](docs/_static/analogvnn_model.png)
 
-[//]: # (![3 Layered Linear Photonic Analog Neural Network]&#40;https://github.com/Vivswan/AnalogVNN/raw/v1.0.0/docs/_static/analogvnn_model.png&#41;)
+[//]: # (![3 Layered Linear Photonic Analog Neural Network]&#40;https://github.com/Vivswan/AnalogVNN/raw/release/docs/_static/analogvnn_model.png&#41;)
 
 ## Abstract
 

diff --git a/analogvnn/parameter/PseudoParameter.py b/analogvnn/parameter/PseudoParameter.py
@@ -178,6 +178,16 @@ def grad(self):
 
         return self._transformed.grad
 
+    @grad.setter
+    def grad(self, grad: Tensor):
+        """Sets the gradient of the parameter.
+
+        Args:
+            grad (Tensor): the gradient.
+        """
+
+        self._transformed.grad = grad
+
     @property
     def module(self):
         """Returns the module.

diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ py-modules = ['analogvnn']
 [project]
 # $ pip install analogvnn
 name = "analogvnn"
-version = "1.0.0"
+version = "1.0.1"
 description = "A fully modular framework for modeling and optimizing analog/photonic neural networks"  # Optional
 readme = "README.md"
 requires-python = ">=3.7"

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
---extra-index-url https://download.pytorch.org/whl/cu116
+--extra-index-url https://download.pytorch.org/whl/cu118
 torch
-torchvision~=0.14.0
+torchvision
 torchaudio
 numpy
 scipy
@@ -11,5 +11,6 @@ importlib-metadata; python_version < '3.8'
 tensorflow>=2.0.0
 tensorboard>=2.0.0
 torchinfo
-# conda install graphviz
+# conda install graphviz python-graphviz pydot pydotplus python-dotenv
+# conda install --channel conda-forge pygraphviz
 graphviz