Update modeling_autoencoder.py
Browse files- modeling_autoencoder.py +0 -2
modeling_autoencoder.py
CHANGED
|
@@ -307,7 +307,6 @@ class AutoEncoder(PreTrainedModel):
|
|
| 307 |
for layer in self.encoder:
|
| 308 |
if isinstance(layer, nn.LSTM):
|
| 309 |
input_ids, (h_n, c_n) = layer(input_ids)
|
| 310 |
-
print(input_ids.size())
|
| 311 |
|
| 312 |
elif isinstance(layer, nn.RNN) or isinstance(layer, nn.GRU):
|
| 313 |
input_ids, h_o = layer(input_ids)
|
|
@@ -320,7 +319,6 @@ class AutoEncoder(PreTrainedModel):
|
|
| 320 |
for layer in self.decoder:
|
| 321 |
if isinstance(layer, nn.LSTM):
|
| 322 |
input_ids, (h_n, c_n) = layer(input_ids)
|
| 323 |
-
print(input_ids.size())
|
| 324 |
|
| 325 |
elif isinstance(layer, nn.RNN) or isinstance(layer, nn.GRU):
|
| 326 |
input_ids, h_o = layer(input_ids)
|
|
|
|
| 307 |
for layer in self.encoder:
|
| 308 |
if isinstance(layer, nn.LSTM):
|
| 309 |
input_ids, (h_n, c_n) = layer(input_ids)
|
|
|
|
| 310 |
|
| 311 |
elif isinstance(layer, nn.RNN) or isinstance(layer, nn.GRU):
|
| 312 |
input_ids, h_o = layer(input_ids)
|
|
|
|
| 319 |
for layer in self.decoder:
|
| 320 |
if isinstance(layer, nn.LSTM):
|
| 321 |
input_ids, (h_n, c_n) = layer(input_ids)
|
|
|
|
| 322 |
|
| 323 |
elif isinstance(layer, nn.RNN) or isinstance(layer, nn.GRU):
|
| 324 |
input_ids, h_o = layer(input_ids)
|