Closed iamyf closed 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 |