Oneflow-Inc / OneFlow-Benchmark

OneFlow models for benchmarking.
104 stars 31 forks source link

add benchmark for alexnet #38

Closed iamyf closed 4 years ago

iamyf commented 4 years ago
改动详情 相关文件
修改alexnet 中卷积层和全连接层的初始化方式 alexnet_model.py
修改训练入口,添加时间打印和选择gpu of_cnn_train_val.py
修改oneflow.function->oneflow.global_function of_cnn_train_val.py, of_cnn_val.py
alexnet训练脚本 of_cnn_train.sh
修改README,添加Alexnet运行说明 README.md