Training Issues!

[sree3333]

Segmentation Models: using keras framework.
INFO: Using device cuda
Traceback (most recent call last):
File "train.py", line 169, in
net = smp.EfficientUnetPlusPlus(encoder_name=args.encoder, encoder_weights="imagenet", in_channels=3, classes=n_classes)
AttributeError: module 'segmentation_models' has no attribute 'EfficientUnetPlusPlus'

[Diacod-I]

It seems the issue is caused because you are using the segmentation_models library module which uses Keras as a framework, instead of segmentation_models_pytorch...

The keras framework doesn't have EfficientUnetPlusPlus. The following command should help you install the right framework.

$ pip install -U git+https://github.com/jlcsilva/segmentation_models.pytorch

[Swiffany]

anyone else have this problem?