Thanks.
I know this way,but is there any way in Pytorch that I can directly load a self-defined class model in new session without define the class in both session? It seems that if I don’t define the class in new session, and then load it, I will get AttributionError : No module named simple in this example.