Skip to content
 
 

Repository files navigation

项目名称

[FNN Training Tools] - [灵活地用参数控制打包好的功能,训练需要的神经网络,目前只有FNN]

 

项目介绍

直接在项目下创建训练模型的代码,复制粘贴两个模板,按需修改。

 

目录结构说明

TRAIN_TOOLS/ ├── init.py ├── basic_steps.py # 一些不同程度打包好的基础步骤,从准备设备,数据集,到训练模型,按需导入需要的功能 ├── my_models.py # 储存常用的神经网络模型或灵活定义神经网络的功能 ├── templates_of_batch_copies.py # 多复本训练流程模板,复制粘贴使用 ├── templates_of_one_batch_simulations.py # 单复本训练流程模板,复制粘贴使用 ├── tools.py # 自定义的方便函数 └── README.md # 项目说明文档(当前文件)  

 

前置准备

  1. conda环境准备
  • 安装 [anaconda],打开[anaconda_prompt]
  • 创建项目环境:
  • conda create [环境名] python=3.9
  • 进入:
  • conda activate [环境名]
  1. 安装依赖库: -conda install torch torchview numpy matplotlib  

  2. 克隆项目到本地:

About

主要基于用minist数据集训练神经网络,目前是简单的全连接神经网络FNN。

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages