如何修改在pytorch中导入的auto-keras模型

时间:2019-05-13 00:47:00

标签: deep-learning computer-vision auto-keras

我正在尝试修改从autokeras模型导入的pytorch容器模型。

cnn_module = pickle_from_file(os.path.join(p, 'best_mnist')) 
model = cnn_module.graph.produce_model()

我想删除最后一层以从完全连接的层获取要素图。但是我无法修改模型。任何人都可以帮忙。谢谢.. !!

0 个答案:

没有答案