Error: 'dict' object has no attribute 'state_dict'"

[ajulyav]

Hello,

I keep having a problem:

" data = self.model.state_dict() if self.model is not None else OrderedDict()
AttributeError: 'dict' object has no attribute 'state_dict'"

ServerRunner - ERROR - [run=2]: Aborting current RUN due to FATAL_SYSTEM_ERROR received: cannot create state_dict from model object

I use the version NV 2.0 and given example of Prostate with DITTO learner, just use a classification dataset and EfficientNetBN to do classification.

In my config of the server:

     "id": "persistor",
      "name": "PTFileModelPersistor",
      "args": {
      "model": {
            "path": "monai.networks.nets.EfficientNetBN",
            "args": {
              "model_name": "efficientnet-b7",
              "spatial_dims": 2,
              "in_channels": 1,
              "num_classes": 8
            }
          }
      }

What could be a problem?

Thank you!