代码收藏家技术教程 2024-12-18

Python 神经网络项目常用语法

基础

1. 导入模块和包

2. 修改系统路径 (sys.path.append)

3. 命令行参数解析 (argparse 模块)

4. 字符串格式化

5. 获取设备

6.设置推理结果的子路径

7. main() 脚本入口点

一、常用函数

1. 学习率调度器

二、类相关

1. 类的定义及初始化

2. 类的实例化及函数调用

三、神经网络常用类

1. 工具类

1.1 正弦位置编码类

1.2 上采样/下采样

1.3 标准化

1.4 归一化层

1.5 提取函数 extract

2. 构建网络块类

2.1 block()

2.2 残差连接

2.3 ResnetBlock

2.3 Attention

基础

import os
import argparse
import sys as s
from accelerate import Accelerator

import：用于导入模块和包，可以选择导入单个模块、多个模块。

from ... import ...：从特定模块中导入具体的类、函数或变量。

impoert ... as ...：可以为导入的模块指定别名，使代码简洁。

作者：一杯水果茶！

分享到：