Spaces:
Sleeping
Sleeping
Update Decoder.py
Browse files- Decoder.py +1 -0
Decoder.py
CHANGED
|
@@ -5,6 +5,7 @@ from torch.nn import functional as F
|
|
| 5 |
from nn_layers import linear_module, location_layer
|
| 6 |
from utils import get_mask_from_lengths
|
| 7 |
|
|
|
|
| 8 |
|
| 9 |
class AttentionNet(nn.Module):
|
| 10 |
# 1024, 512, 128, 32, 31
|
|
|
|
| 5 |
from nn_layers import linear_module, location_layer
|
| 6 |
from utils import get_mask_from_lengths
|
| 7 |
|
| 8 |
+
torch.manual_seed(1234)
|
| 9 |
|
| 10 |
class AttentionNet(nn.Module):
|
| 11 |
# 1024, 512, 128, 32, 31
|