这里对改进sample code中的模型以及引用其他模型进行了尝试,但效果不佳。

如果你想对sample code中的模型进行改进,直接在类Classifier中的self.cnn以及self.fc模块中进行修改即可

如果你想引入其他的模型,可以参考下面的代码