Update train.py
Browse files
train.py
CHANGED
|
@@ -52,8 +52,7 @@ class NormalizerImageClassification(Normalizer):
|
|
| 52 |
d_model = 256,
|
| 53 |
d_ffn = 512,
|
| 54 |
num_tokens = 64,
|
| 55 |
-
num_layers = 4
|
| 56 |
-
|
| 57 |
):
|
| 58 |
super().__init__(d_model, d_ffn, num_tokens, num_layers)
|
| 59 |
self.patcher = nn.Conv2d(
|
|
|
|
| 52 |
d_model = 256,
|
| 53 |
d_ffn = 512,
|
| 54 |
num_tokens = 64,
|
| 55 |
+
num_layers = 4
|
|
|
|
| 56 |
):
|
| 57 |
super().__init__(d_model, d_ffn, num_tokens, num_layers)
|
| 58 |
self.patcher = nn.Conv2d(
|