标签: 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()
我想删除最后一层以从完全连接的层获取要素图。但是我无法修改模型。任何人都可以帮忙。谢谢.. !!