Skip to content

Commit e62ebed

Browse files
committed
修正session的打印值
1 parent 0c3cbb4 commit e62ebed

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: deeplog/deeplog.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,8 @@ def generate(self, name, window_size, local):
5555
outputs.append(line[i+window_size])
5656
line = self.readline(local)
5757
num_sessions += 1
58-
logger.info('Number of session({}): {}'.format(name, len(inputs)))
58+
# logger.info('Number of session({}): {}'.format(name, len(inputs)))
59+
logger.info('Number of session({}): {}'.format(name, num_sessions))
5960
logger.info('Number of seqs({}): {}'.format(name, len(inputs)))
6061
dataset = TensorDataset(torch.tensor(inputs, dtype=torch.float), torch.tensor(outputs))
6162
return dataset

0 commit comments

Comments
 (0)