Fix auto_map + sync source files
Browse files- modeling_rtmdet.py +2 -0
modeling_rtmdet.py
CHANGED
|
@@ -1797,6 +1797,8 @@ class RTMDetModel(PreTrainedModel):
|
|
| 1797 |
|
| 1798 |
# Initialize weights
|
| 1799 |
self.init_weights()
|
|
|
|
|
|
|
| 1800 |
|
| 1801 |
def init_weights(self):
|
| 1802 |
"""Initialize the weights of the model."""
|
|
|
|
| 1797 |
|
| 1798 |
# Initialize weights
|
| 1799 |
self.init_weights()
|
| 1800 |
+
# Required: triggers post_init() which sets all_tied_weights_keys etc.
|
| 1801 |
+
self.post_init()
|
| 1802 |
|
| 1803 |
def init_weights(self):
|
| 1804 |
"""Initialize the weights of the model."""
|