plant_omni does not have dim=3

[donghyunkm]

Hello,

As I was going through the notebooks, I found an error with the 'plant_omni" pretrained model.

The code and this link indicates that 'plant_omni' should have dim=3 but using dim=3 results in "failed to load model Error(s) in loading state_dict for CPnet" and additional errors during segmentation.

When I create a 'plant_omni' model without the dim argument, the model is shown to have model.dim = 2.

Thus, I believe the checkpoint used to internally load the pretrained model may be incorrect/outdated?

Attached are relevant screenshots.

Thank you for your help!