Take() got an unexpected keyword argument 'count'.

[0x5c4r3]

Heya, I'm trying to use the model to perform some image separation using the mnist database. I literally copy your code and modify some parameters to match with the mnist database, but I get this weird mesage:

If you're wondering, here's how I create the train and test dataset:

(mnist_x_train, mnist_y_train), (mnist_x_test, mnist_y_test) = mnist.load_data()

Not sure what this is related to, any ideas?

Thanks, Mat

[jakeret]

Hi Mat, sorry for the very late reply. Could it be, that you are trying to run unet with an older version of Tensorflow? The stable tf.data.Dataset implementation has the keyword count.