Skip to content
Snippets Groups Projects
Commit bf4b668a authored by André Anjos's avatar André Anjos 💬
Browse files

[models.model] Explicitly set the datamodule batch-size to 1 to avoid issues...

[models.model] Explicitly set the datamodule batch-size to 1 to avoid issues when running in an combo-experiment
parent 812e23df
No related branches found
No related tags found
1 merge request!49Optimise device allocation for all attributes used in the training loop
Pipeline #88041 canceled
......@@ -180,6 +180,7 @@ def generate(
datamodule.cache_samples = cache_samples
datamodule.parallel = parallel
datamodule.model_transforms = model.model_transforms
datamodule.batch_size = 1
datamodule.prepare_data()
datamodule.setup(stage="predict")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment