Upload modeling_normwear.py
Browse files- modeling_normwear.py +2 -0
modeling_normwear.py
CHANGED
|
@@ -13,6 +13,8 @@ class NormWear2Model(PreTrainedModel):
|
|
| 13 |
|
| 14 |
def __init__(self, config: NormWear2Config):
|
| 15 |
super().__init__(config)
|
|
|
|
|
|
|
| 16 |
|
| 17 |
self.normwear = NormWear2(
|
| 18 |
patch_size=config.patch_size,
|
|
|
|
| 13 |
|
| 14 |
def __init__(self, config: NormWear2Config):
|
| 15 |
super().__init__(config)
|
| 16 |
+
|
| 17 |
+
self.patch_size = config.patch_size
|
| 18 |
|
| 19 |
self.normwear = NormWear2(
|
| 20 |
patch_size=config.patch_size,
|