It doesn't work while fitting.

[laetokang]

While I was running BERT, suddenly I got the following error message.
'''
inputs, attrs, num_outputs)
tensorflow.python.framework.errors_impl.InvalidArgumentError: No OpKernel was registered to support Op 'NcclAllReduce' used by {{node while/body/_1/AdamWeightDecay/NcclAllReduce}} with these attrs: [reduction="sum", shared_name="c1", T=DT_FLOAT, num_devices=2]
Registered devices: [CPU, GPU, XLA_CPU, XLA_GPU]
Registered kernels:

     [[while/body/_1/AdamWeightDecay/NcclAllReduce]] [Op:__inference_train_steps_46285]

'''
I don't know what does it mean. Please Help..

[artitw]

What’s your setup? Did you get the colab notebook demo to work?

[artitw]

Please take a look at https://github.com/artitw/text2text and ask questions there if necessary.