Keras实施“广泛的残留网络”
此存储库包含使用Keras运行广泛残差网络的代码。
论文(v1): : (此后,作者发布了该论文的v2,该版本引入了稍有不同的预处理,并稍微提高了精度)。
原始代码: :
依存关系:
pip install -r requirements.txt
要绘制所使用模型的架构(如绘制的WRN-16-2架构图),您需要安装pydot和graphviz 。 我建议使用conda install -c conda-forge python-graphviz :
培训细节:
使用以下命令运行默认配置(即,从原始纸张/代码获得的CIFAR10的最佳配置,WRN-28-10,没有丢失):
$ python main.py
main.py顶部有三个配置部分:
:包含数据详细信息。
:包括作者试验的主要参数。
:定义有关在何处保存模型/检查点权重和图
评论0
最新资源