As an addition to what @G.M said, could you please print the complete stack trace you get from the NotImplementedError?
You would get this error, e.g. if you have a typo in a method name (e.g. forwrad instead of forward) or if you use a wrong indentation.