Abdullah-Nazhat
/

Lite_Tensor_Mapper

Model card Files Files and versions

Abdullah-Nazhat commited on Jul 2

Commit

01ba50b

·

verified ·

1 Parent(s): 3618e08

Update train.py

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -65,7 +65,7 @@ class LiteTensorMapperImageClassification(LiteTensorMapper):
     ):
-        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
         )