Abdullah-Nazhat
/

Tensor_Mapper

Model card Files Files and versions

Abdullah-Nazhat commited on Jul 2

Commit

efad028

·

verified ·

1 Parent(s): b779692

Update train.py

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -63,7 +63,7 @@ class TensorMapperImageClassification(TensorMapper):
     ):
-        super().__init__(d_model, num_patches,num_layers)
         self.patcher = nn.Conv2d(
             in_channels, d_model, kernel_size=patch_size, stride=patch_size
         )

     ):
+        super().__init__(d_model, num_layers)
         self.patcher = nn.Conv2d(
             in_channels, d_model, kernel_size=patch_size, stride=patch_size
         )