参考我之前写的博文:Centos7 安装Tensorflow2.1 GPU以及Pytorch1.3 GPU(CUDA10.1)
强烈不建议混用,即使两个模型的名称结构完全一致也不要混用,里面有坑,用什么方法训练的模型就载入相应的模型权重
subclassed模型在实例化时没有自动进行build操作(只有在开始训练时,才会自动进行build),如果需要使用summary操作,需要提前手动build
model.build((batch_size, height, width, channel))
- pip install pydot==1.2.3
- sudo apt-get install graphviz
- pip install pydot==1.2.3
- 安装graphviz,并添加相关环境变量
参考连接:XifengGuo/CapsNet-Keras#7
如果不清除历史梯度,就会对计算的历史梯度进行累加(通过这个特性你能够变相实现一个很大batch数值的训练)
参考链接:https://www.zhihu.com/question/303070254
pillow版本过高导致,安装版本号小于7.0.0即可