MORTM 2.0b3__tar.gz → 2.0b4__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {mortm-2.0b3 → mortm-2.0b4}/MORTM.egg-info/PKG-INFO +1 -1
- {mortm-2.0b3 → mortm-2.0b4}/PKG-INFO +1 -1
- {mortm-2.0b3 → mortm-2.0b4}/mortm/train.py +4 -3
- {mortm-2.0b3 → mortm-2.0b4}/setup.py +1 -1
- {mortm-2.0b3 → mortm-2.0b4}/MORTM.egg-info/SOURCES.txt +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/MORTM.egg-info/dependency_links.txt +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/MORTM.egg-info/top_level.txt +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/README.md +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/PositionalEncoding.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/RelativePositionalRepresentations.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/__init__.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/constants.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/convert.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/datasets.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/de_convert.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/eval.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/generate.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/gmail_messanger.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/loss.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/mask.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/messager.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/mortm.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/noam.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/progress.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/reinforcement.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/rpr.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/token.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/mortm/tokenizer.py +0 -0
- {mortm-2.0b3 → mortm-2.0b4}/setup.cfg +0 -0
|
@@ -95,9 +95,10 @@ def _get_padding_mask(input_ids, progress: LearningProgress):
|
|
|
95
95
|
|
|
96
96
|
def collate_fn(batch):
|
|
97
97
|
# バッチ内のテンソルの長さを揃える(パディングする)
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
98
|
+
src_list = [item[0] for item in batch] # 各タプルのsrcを抽出
|
|
99
|
+
tgt_list = [item[1] for item in batch] # 各タプルのtgtを抽出
|
|
100
|
+
src = pad_sequence(src_list, batch_first=True, padding_value=0)
|
|
101
|
+
tgt = pad_sequence(tgt_list, batch_first=True, padding_value=0)
|
|
101
102
|
return (src, tgt)
|
|
102
103
|
|
|
103
104
|
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|