From fef83344a653ce4084d8f0d9276b4c6292ddb986 Mon Sep 17 00:00:00 2001 From: Mathijs de Boer Date: Mon, 4 Dec 2023 14:35:50 +0100 Subject: [PATCH] Forgot to write the new config --- nnunetv2/model_sharing/onnx_export.py | 23 +---------------------- 1 file changed, 1 insertion(+), 22 deletions(-) diff --git a/nnunetv2/model_sharing/onnx_export.py b/nnunetv2/model_sharing/onnx_export.py index d07a52dbb..73ce6dae0 100644 --- a/nnunetv2/model_sharing/onnx_export.py +++ b/nnunetv2/model_sharing/onnx_export.py @@ -158,28 +158,7 @@ def export_onnx_model( } json.dump( - { - "dataset_name": dataset_name, - "configuration": c, - "fold": fold, - "checkpoint_name": checkpoint_name, - "configuration_manager": { - k: config.configuration[k] - for k in [ - "patch_size", - "spacing", - "normalization_schemes", - # These are mostly interesting for certification - # uses, but they are also useful for debugging. - "UNet_class_name", - "UNet_base_num_features", - "unet_max_num_features", - "conv_kernel_sizes", - "pool_op_kernel_sizes", - "num_pool_per_axis", - ] - }, - }, + config_dict, f, indent=4, )