Skip to content

Commit 24a9d18

Browse files
committed
fix codestyle
1 parent c587dba commit 24a9d18

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

models/recall/mind/mind_reader.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,10 @@
1616
import numpy as np
1717
from paddle.io import IterableDataset
1818
import random
19+
1920
random.seed(12345)
2021

22+
2123
class RecDataset(IterableDataset):
2224
def __init__(self, file_list, config):
2325
super(RecDataset, self).__init__()

models/recall/mind/net.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,10 @@ def sequence_mask(self, lengths, maxlen=None, dtype="bool"):
167167
batch_size = paddle.shape(lengths)[0]
168168
if maxlen is None:
169169
maxlen = lengths.max()
170-
row_vector = paddle.arange(0, maxlen, 1).unsqueeze(0).expand(
171-
shape=(batch_size, maxlen)).reshape((batch_size, -1, maxlen))
170+
row_vector = paddle.arange(
171+
0, maxlen,
172+
1).unsqueeze(0).expand(shape=(batch_size, maxlen)).reshape(
173+
(batch_size, -1, maxlen))
172174
lengths = lengths.unsqueeze(-1)
173175
mask = row_vector < lengths
174176
return mask.astype(dtype)

0 commit comments

Comments
 (0)