Skip to content
Snippets Groups Projects
Commit d8242837 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

[classification.scripts] Fix call to setup_datamodule

parent c6eac964
No related branches found
No related tags found
1 merge request!46Create common library
Hide whitespace changes
Inline Side-by-side
src/mednet/libs/classification/scripts/train.py
+ 1
− 0
View file @ d8242837
...@@ -75,6 +75,7 @@ def train( ...@@ -75,6 +75,7 @@ def train(
setup_datamodule( setup_datamodule(
datamodule, datamodule,
model, model,
batch_size,
drop_incomplete_batch, drop_incomplete_batch,
cache_samples, cache_samples,
parallel, parallel,
......
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