__call__() got an unexpected keyword argument 'training'

This error message typically occurs when using a class that inherits from a parent class with a __call__() method that does not accept a keyword argument named 'training'.

The __call__() method is a special method in Python that allows an object to be called like a function. It is commonly used in deep learning frameworks such as PyTorch and TensorFlow to define the forward pass of a neural network.

To resolve this error, you should check the definition of the parent class and ensure that the __call__() method does not accept a keyword argument called 'training'. If it does, you can either remove the keyword argument from the parent class or modify the child class to remove the argument when calling the parent's __call__() method.