diff --git a/decloud/production/inference.py b/decloud/production/inference.py index d2d4e5b7868d4ad2487030696c27cc7c18a5154c..4570577681b0f6beed242c335c53fdd1a6bf7909 100644 --- a/decloud/production/inference.py +++ b/decloud/production/inference.py @@ -86,7 +86,7 @@ def inference(sources, sources_scales, pad, ts, savedmodel_dir, out_tensor, out_ get_key("placeholder"): placeholder}) # Model - parameters.update({"model.dir": savedmodel_dir, "model.fullyconv": 1, + parameters.update({"model.dir": savedmodel_dir, "model.fullyconv": True, "output.names": [padded_tensor_name(out_tensor, pad)], "output.efieldx": efield, "output.efieldy": efield, "optim.tilesizex": efield, "optim.tilesizey": efield, diff --git a/decloud/production/monthly_synthesis_processor_s2.py b/decloud/production/monthly_synthesis_processor_s2.py index fe2423d5829fbe927ee7c1d5ea2e2d7a37eb2f26..b0624143952b17db70ec6c8b8a527878839b7900 100644 --- a/decloud/production/monthly_synthesis_processor_s2.py +++ b/decloud/production/monthly_synthesis_processor_s2.py @@ -109,7 +109,7 @@ def monthly_synthesis_inference(sources, sources_scales, pad, ts, savedmodel_dir get_key("placeholder"): placeholder}) # Model - infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": 1, + infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": True, "output.names": [padded_tensor_name(out_tensor, pad)], "output.efieldx": efield, "output.efieldy": efield, "optim.tilesizex": efield, "optim.tilesizey": efield, diff --git a/decloud/production/monthly_synthesis_processor_s2s1.py b/decloud/production/monthly_synthesis_processor_s2s1.py index f69dbc15c94d32e2bb3e00256b1987b3206a91c1..46e10e7e450404c2241bc3e6179f6762372bd603 100644 --- a/decloud/production/monthly_synthesis_processor_s2s1.py +++ b/decloud/production/monthly_synthesis_processor_s2s1.py @@ -109,7 +109,7 @@ def monthly_synthesis_inference(sources, sources_scales, pad, ts, savedmodel_dir get_key("placeholder"): placeholder}) # Model - infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": 1, + infer_params.update({"model.dir": savedmodel_dir, "model.fullyconv": True, "output.names": [padded_tensor_name(out_tensor, pad)], "output.efieldx": efield, "output.efieldy": efield, "optim.tilesizex": efield, "optim.tilesizey": efield,