Abdullah-Nazhat
/

Tensor_Mapper

Model card Files Files and versions

Abdullah-Nazhat commited on Aug 25

Commit

f69ca2c

·

verified ·

1 Parent(s): da194f2

Update train.py

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -50,7 +50,7 @@ class TensorMapperImageClassification(TensorMapper):
         in_channels = 3,
         num_classes = 10,
         d_model = 256,
-        num_layers = 4,
     ):
         super().__init__(d_model, num_layers)

         in_channels = 3,
         num_classes = 10,
         d_model = 256,
+        num_layers = 4
     ):
         super().__init__(d_model, num_layers)