zhaoyiming / densenet_mindspore

implement densenet in mindspore

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

model.py : DenseNet网络模型文件(k=12, Depth=40)

model_bc.py : DenseNet-BC网络模型文件(k=12, Depth=100)

parameter.py : 包含要使用到的超参数和参数

dataset.py : 加载Cifar10数据,需要确定数据集位置

train.py : 训练文件

eval.py : 推理文件,需要确定模型位置

About

implement densenet in mindspore


Languages

Language:Python 100.0%