Updated layers.py

ivy-llc · Aug 25, 2023 · 53ef379 · 53ef379
1 parent adf548e
commit 53ef379
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/ivy_models/regnet/layers.py b/ivy_models/regnet/layers.py
@@ -189,7 +189,7 @@ def _scale(self, input: ivy.Array) -> ivy.Array:
         scale = self.fc2(scale)
         return self.scale_activation(scale)
 
-    def forward(self, input: ivy.Array) -> ivy.Array:
+    def _forward(self, input: ivy.Array) -> ivy.Array:
         scale = self._scale(input)
         return scale * input
 

diff --git a/scratch.py b/scratch.py
@@ -1,9 +1,6 @@
 from ivy_models.helpers import load_torch_weights
 from ivy_models.regnet.regnet import RegNet
 from ivy_models.regnet.layers import BlockParams
-import ivy
-
-ivy.set_backend("torch")
 
 # num_classes = 1000
 # dropout = 0.5