ivadomed / model_seg_ms_mp2rage

Model repository for MS lesion segmentation on MP2RAGE data from University of Basel
MIT License
2 stars 0 forks source link

Explain model size differences #66

Open jcohenadad opened 1 year ago

jcohenadad commented 1 year ago

Uzay's model was 6.32 MB Mine was 30 MB.

Some comparisons:

Uzay:

   "Modified3DUNet": {
        "applied": true,
        "length_3D": [
            128,
            64,
            32
        ],
        "stride_3D": [
            128,
            64,
            32
        ],
        "n_filters": 8
        "CenterCrop": {
            "size": [
                128,
                64,
                32
            ]
        },

Mine:

        "length_3D": [32, 64, 64],
        "stride_3D": [1, 1, 64],
        "n_filters": 16
        "CenterCrop": {"size": [32, 64, 128]},

So, the centercrop is the same, the stride differs a lot, and n_filter is twice as large in my model.