AttributeError: 'CNNModel' object has no attribute 'feat_extract'

[Lxltxpku]

Hello, I encountered a problem while executing tiatoolbox/examples/05-patch-prediction.ipynb:

AttributeError: 'CNNModel' object has no attribute 'feat_extract'. 

All the previous steps ran normally.

Here is the detailed information:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[9], line 2
      1 predictor = PatchPredictor(pretrained_model="resnet18-kather100k", batch_size=32)
----> 2 output = predictor.predict(imgs=patch_list, mode="patch", device=device)

File /data/data_mailab015/env_set/anaconda3/envs/tiatoolbox-dev/lib/python3.9/site-packages/tiatoolbox/models/engine/patch_predictor.py:912, in PatchPredictor.predict(self, imgs, masks, labels, mode, ioconfig, patch_input_shape, stride_shape, resolution, units, device, return_probabilities, return_labels, merge_predictions, save_dir, save_output)
    908     raise ValueError(
    909         msg,
    910     )
    911 if mode == "patch":
--> 912     return self._predict_patch(
    913         imgs,
    914         labels,
    915         return_probabilities=return_probabilities,
    916         return_labels=return_labels,
    917         device=device,
    918     )
    920 if not isinstance(imgs, list):
    921     msg = "Input to `tile` and `wsi` mode must be a list of file paths."

File /data/data_mailab015/env_set/anaconda3/envs/tiatoolbox-dev/lib/python3.9/site-packages/tiatoolbox/models/engine/patch_predictor.py:640, in PatchPredictor._predict_patch(self, imgs, labels, device, return_probabilities, return_labels)
    638 return_coordinates = False
    639 dataset = PatchDataset(imgs, labels)
--> 640 return self._predict_engine(
...
    feat = self.feat_extract(imgs)
  File "/data/data_mailab015/env_set/anaconda3/envs/tiatoolbox-dev/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1931, in __getattr__
    raise AttributeError(
AttributeError: 'CNNModel' object has no attribute 'feat_extract' 

[shaneahmed]

Please can you share

• TIAToolbox version
• Your operating system name and version.
• Any details about your local setup that might be helpful in troubleshooting.
• Detailed steps to reproduce the bug.