Skip to content

Commit

Permalink
update call regarding to gp.pad
Browse files Browse the repository at this point in the history
  • Loading branch information
mzouink committed Feb 14, 2024
1 parent a33a5e3 commit e20d8a9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions dacapo/experiments/trainers/gunpowder_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,9 +133,9 @@ def build_batch_provider(self, datasets, model, task, snapshot_container=None):
mask_placeholder,
drop_channels=True,
)
+ gp.Pad(raw_key, None, 0)
+ gp.Pad(gt_key, None, 0)
+ gp.Pad(mask_key, None, 0)
+ gp.Pad(raw_key, None)
+ gp.Pad(gt_key, None)
+ gp.Pad(mask_key, None)
+ gp.RandomLocation(
ensure_nonempty=sample_points_key
if points_source is not None
Expand Down

0 comments on commit e20d8a9

Please sign in to comment.