segmentation_models icon indicating copy to clipboard operation
segmentation_models copied to clipboard

AttributeError: 'Model' object has no attribute '_in_multi_worker_mode'

Open Anand2805 opened this issue 5 years ago • 2 comments
trafficstars

import segmentation_models as sm from segmentation_models.metrics import iou_score from segmentation_models import Unet

import tensorflow as tf tf.keras.backend.set_image_data_format('channels_last')

model = Unet('resnet34', encoder_weights='imagenet', classes=40, activation='sigmoid') model.summary()

optim = tf.keras.optimizers.Adam(0.0001)

focal_loss = sm.losses.cce_dice_loss

model.compile(optim, focal_loss, metrics=[iou_score])

callbacks = [ tf.keras.callbacks.ModelCheckpoint('model/best_model.h5', save_weights_only=True, save_best_only=True,
mode='min', monitor='val_iou_score'), tf.keras.callbacks.ReduceLROnPlateau(monitor='val_iou_score', min_lr=0.000001,patience=2), ]

image

im getting the above error while training the model.

Anand2805 avatar Jul 13 '20 11:07 Anand2805

import segmentation_models as sm from segmentation_models.metrics import iou_score from segmentation_models import Unet

import tensorflow as tf tf.keras.backend.set_image_data_format('channels_last')

model = Unet('resnet34', encoder_weights='imagenet', classes=40, activation='sigmoid') model.summary()

optim = tf.keras.optimizers.Adam(0.0001)

focal_loss = sm.losses.cce_dice_loss

model.compile(optim, focal_loss, metrics=[iou_score])

callbacks = [ tf.keras.callbacks.ModelCheckpoint('model/best_model.h5', save_weights_only=True, save_best_only=True, mode='min', monitor='val_iou_score'), tf.keras.callbacks.ReduceLROnPlateau(monitor='val_iou_score', min_lr=0.000001,patience=2), ]

image

im getting the above error while training the model.

Hey, could you solve this?

AbtinDjavadifar avatar Aug 13 '20 05:08 AbtinDjavadifar

For the moment, I am solving this with

model._in_multi_worker_mode = lambda: None

eddienko avatar Dec 16 '20 10:12 eddienko