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(
setup_datamodule(
datamodule,
model,
batch_size,
drop_incomplete_batch,
cache_samples,
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