Skip to content

Commit

Permalink
training regressions in transcription/segmentation
Browse files Browse the repository at this point in the history
  • Loading branch information
@mittagessen
mittagessen committed Nov 24, 2023
1 parent c1f5a7d commit 74e23a3
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions kraken/lib/train.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -357,7 +357,7 @@ def __init__(self,
height,
width,
channels,
(self.hparams.pad, 0),
(self.hparams.hyper_params['pad'], 0),
valid_norm,
force_binarization)

Expand Down Expand Up @@ -782,7 +782,7 @@ def __init__(self,
height,
width,
channels,
self.hparams.hyper_params.padding,
self.hparams.hyper_params['padding'],
valid_norm=False,
force_binarization=force_binarization)

Expand All @@ -809,10 +809,10 @@ def __init__(self,
merge_baselines = None

train_set = BaselineSet(training_data,
line_width=self.hparams.hyper_params.line_width,
line_width=self.hparams.hyper_params['line_width'],
im_transforms=transforms,
mode=format_type,
augmentation=self.hparams.hyper_params.augment,
augmentation=self.hparams.hyper_params['augment'],
valid_baselines=valid_baselines,
merge_baselines=merge_baselines,
valid_regions=valid_regions,
Expand All @@ -824,7 +824,7 @@ def __init__(self,

if evaluation_data:
val_set = BaselineSet(evaluation_data,
line_width=self.hparams.hyper_params.line_width,
line_width=self.hparams.hyper_params['line_width'],
im_transforms=transforms,
mode=format_type,
augmentation=False,
Expand Down

0 comments on commit 74e23a3

Please sign in to comment.