Paweł Gadziński

Unfortunately removing extra_state key from the dictionary will result with error when `load_state_dict(..., strict=True)`. I see no way of doing that without breaking other thing. I modified the test, instead...

But this tutorial mode change is good. Should I remove it and add to the new PR?