Skip to content

Commit

Permalink
check if batch size exists for array prediction"
Browse files Browse the repository at this point in the history
  • Loading branch information
Benjamin Morris committed Sep 20, 2024
1 parent 6998235 commit 1f8f705
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion cyto_dl/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ def train(cfg: DictConfig, data=None) -> Tuple[dict, dict]:
log.info(f"Instantiating data <{cfg.data.get('_target_', cfg.data)}>")

use_batch_tuner = False
if cfg.data.batch_size == "AUTO":
if cfg.data.get("batch_size") == "AUTO":
use_batch_tuner = True
cfg.data.batch_size = 1

Expand Down

0 comments on commit 1f8f705

Please sign in to comment.