Commit 5401d462 authored by SPeillet's avatar SPeillet
Browse files

ENH: fix optional epochs for m3fusion

parent b63e5ce7
......@@ -752,7 +752,9 @@ def main(argv):
deepTraining([samples_fld + '/GT_samples.shp'],cfield,model_fld,params,var_list)
elif params[1].lower() == 'm3fusion':
import Moringa2DL
epochs=int(params[2])
epochs = 100
if len(params)>1 :
epochs = int(params[2])
tf_data = Moringa2DL.generateAll(feat_fld,reference,seg_src,cfield,ofld=samples_fld)
if ch_mode == 0 or ch_mode == 2 :
tf_valid_data = Moringa2DL.generateAll(feat_fld,validation,seg_src,cfield,ofld=val_fld,dstype='valid')
......
Markdown is supported
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