minor fixes

plutonium-239 · Apr 13, 2024 · 121b5ff · 121b5ff
1 parent 7826b8c
commit 121b5ff
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 10 deletions.
diff --git a/experiments/paper_demo.py b/experiments/paper_demo.py
@@ -99,6 +99,7 @@
     architecture,
     vjp_improvements,
     cases,
+    'results'
 )
 
 for model in models:

diff --git a/memsave_torch/nn/Conv1d.py b/memsave_torch/nn/Conv1d.py
@@ -5,7 +5,7 @@
 
 import torch
 import torch.nn as nn
-from Conv2d import _MemSaveConv
+from memsave_torch.nn.Conv2d import _MemSaveConv
 
 
 class MemSaveConv1d(nn.Conv1d):

diff --git a/memsave_torch/nn/ReLU.py b/memsave_torch/nn/ReLU.py
@@ -11,15 +11,7 @@ class MemSaveReLU(nn.ReLU):
     """MemSaveReLU."""
 
     def __init__(self):
-        """Inits a MemSaveReLU layer with the given params.
-
-        Args:
-            in_features: in_features
-            out_features: out_features
-            bias: bias
-            device: device
-            dtype: dtype
-        """
+        """Inits a MemSaveReLU layer with the given params."""
         super().__init__()
 
     def forward(self, x):