Can you try without setting dtype to float as string? By default the dtype is set to float32



 Sayan_Nath:

class_head = Dense(NUM_CLASSES, activation="softmax")(class_head)


Also if you can provide a reproducible code example that will be great.