feat: added debug print
Browse files- modeling_bert.py +1 -0
modeling_bert.py
CHANGED
|
@@ -423,6 +423,7 @@ class BertModel(BertPreTrainedModel):
|
|
| 423 |
)
|
| 424 |
|
| 425 |
def to(self, *args, **kwargs):
|
|
|
|
| 426 |
result = super().to(*args, **kwargs)
|
| 427 |
if (len(args) > 0 and isinstance(args[0], torch.dtype)) or "dtype" in kwargs:
|
| 428 |
for layer in result.encoder.layers:
|
|
|
|
| 423 |
)
|
| 424 |
|
| 425 |
def to(self, *args, **kwargs):
|
| 426 |
+
print(f'In BERT, calling to({args, kwargs})')
|
| 427 |
result = super().to(*args, **kwargs)
|
| 428 |
if (len(args) > 0 and isinstance(args[0], torch.dtype)) or "dtype" in kwargs:
|
| 429 |
for layer in result.encoder.layers:
|