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.

[AnshuGusain1]

Hi, I am having the same issue on the latest version of Tensorflow

I create my dataset in the program and then split it into testing and training sets.

They are numpy arrays