Skip to content

Commit ac54712

Browse files
lamblinnotoraptor
authored andcommitted
More small fixes
1 parent 739870e commit ac54712

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

code/fcn_2D_segm/train_fcn8.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def train(dataset, learn_step=0.005,
161161
seq_per_subset=0,
162162
seq_length=0,
163163
data_augm_kwargs=data_augmentation,
164-
return_one_hot=one_hot,
164+
return_one_hot=False,
165165
return_01c=False,
166166
overlap=0,
167167
use_threads=False,
@@ -183,7 +183,7 @@ def train(dataset, learn_step=0.005,
183183
batch_size=batch_size[2],
184184
seq_per_subset=0,
185185
seq_length=0,
186-
return_one_hot=one_hot,
186+
return_one_hot=False,
187187
return_01c=False,
188188
overlap=0,
189189
use_threads=False,

0 commit comments

Comments
 (0)