"docs/git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "6a8ed4a252e653ddd52bd31b3eb53edc1273ce40"
Commit 81ace092 authored by Myle Ott's avatar Myle Ott
Browse files

Fix max_positions calculation in train.py

parent 33469436
...@@ -79,11 +79,11 @@ def main(): ...@@ -79,11 +79,11 @@ def main():
# The max number of positions can be different for train and valid # The max number of positions can be different for train and valid
# e.g., RNNs may support more positions at test time than seen in training # e.g., RNNs may support more positions at test time than seen in training
max_positions_train = (args.max_source_positions, args.max_target_positions) max_positions_train = (
max_positions_valid = (
min(args.max_source_positions, model.max_encoder_positions()), min(args.max_source_positions, model.max_encoder_positions()),
min(args.max_target_positions, model.max_decoder_positions()) min(args.max_target_positions, model.max_decoder_positions())
) )
max_positions_valid = (model.max_encoder_positions(), model.max_decoder_positions())
# Start multiprocessing # Start multiprocessing
trainer = MultiprocessingTrainer(args, model, criterion) trainer = MultiprocessingTrainer(args, model, criterion)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment