Skip to content
Snippets Groups Projects
Commit e007a791 authored by Martyna Wiącek's avatar Martyna Wiącek
Browse files

fixed passing batch size when default model is initialized

parent 54e37314
Branches
Tags
1 merge request!47Fixed multiword prediction + bug that made the code write empty predictions
......@@ -161,7 +161,7 @@ def get_defaults(dataset_reader: Optional[DatasetReader],
)
if not training_data_loader:
training_data_loader = default_data_loader(dataset_reader, training_data_path)
training_data_loader = default_data_loader(dataset_reader, training_data_path, FLAGS.batch_size)
else:
if training_data_path:
training_data_loader.data_path = training_data_path
......@@ -170,7 +170,7 @@ def get_defaults(dataset_reader: Optional[DatasetReader],
str(training_data_loader.data_path), prefix=prefix)
if FLAGS.validation_data_path and not validation_data_loader:
validation_data_loader = default_data_loader(dataset_reader, validation_data_path)
validation_data_loader = default_data_loader(dataset_reader, validation_data_path, FLAGS.batch_size)
elif FLAGS.validation_data_path and validation_data_loader:
if validation_data_path:
validation_data_loader.data_path = validation_data_path
......
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