0%

torch | 如何设计一个扩展性的深度学习项目

我们设计一个深度学习项目,可以满足一下几个特性

  • 可以随时切换 model
  • 通过配置来更改参数
  • 无缝切换

目录


|___checkpoints
|___dataset
|______数据集
|___logger
|______日志
|___main
|______主要目录,包括 入口文件,执行文件,数据载入文件等
|___net
|______网络层,包括各种 model 比如 DNN resnet rnn
|___pre
|______预处理数据,制作 dataset
|___util
|______工具目录
|___config.py
|______配置文件

使用


一般来说写好入口文件之类的,代码就不需要变动了,改变的一般为

  • 增加或者修改 model
  • 修改 config.py 文件
请我喝杯咖啡吧~