Moved linear dictionary to bottom for ordering

This allows dictionary to see classes.
mmoffatt2 · Apr 9, 2024 · 5290cd4 · 5290cd4
1 parent c7fff41
commit 5290cd4
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/variations/linear_variations.py b/variations/linear_variations.py
@@ -2,12 +2,6 @@
 import torch.nn as nn
 import math
 
-linear_dictionary = {
-    "linear": nn.Linear(),
-    "bitlinear": BitLinear(),
-    "bitlinear_optimized": BitLinearOptimized(),
-}
-
 
 class BitLinear(nn.Linear):
     """PyTorch BitLinear Layer
@@ -175,3 +169,10 @@ def forward(self, input):
         output = self.quantize_activations_groupwise(output)
 
         return output
+
+
+linear_dictionary = {
+    "linear": nn.Linear,
+    "bitlinear": BitLinear,
+    "bitlinear_optimized": BitLinearOptimized,
+}