合成学习器

用于快速测试训练循环和回调函数

源代码

synth_dbunch

 synth_dbunch (a=2, b=3, bs=16, n_train=10, n_valid=2, cuda=False)

源代码

RegModel

 RegModel ()

与 nn.Module 相同，但子类无需调用 super().__init__

源代码

synth_learner

 synth_learner (n_trn=10, n_val=2, cuda=False, lr=0.001, data=None,
                model=None, loss_func:Callable|None=None,
                opt_func:Optimizer|OptimWrapper=<function Adam>,
                splitter:Callable=<function trainable_params>,
                cbs:Callback|MutableSequence|None=None,
                metrics:Callable|MutableSequence|None=None,
                path:str|Path|None=None, model_dir:str|Path='models',
                wd:float|int|None=None, wd_bn_bias:bool=False,
                train_bn:bool=True, moms:tuple=(0.95, 0.85, 0.95),
                default_cbs:bool=True)
类型 默认值 详情
n_trn int 10
n_val int 2
cuda bool False
lr float | slice 0.001 默认学习率
data NoneType None
model Callable 用于训练或推理的 PyTorch 模型
loss_func 可选 None 损失函数。默认为 dls 中的损失函数
opt_func fastai.optimizer.Optimizer | fastai.optimizer.OptimWrapper Adam 训练优化函数
splitter Callable trainable_params 将模型分割成参数组。默认为一个参数组
cbs fastai.callback.core.Callback | collections.abc.MutableSequence | None None Callbacks 要添加到 Learner
metrics Union None 在验证集上计算的 Metrics
path str | pathlib.Path | None None 保存、加载和导出模型的父目录。默认为 dls 的 path
model_dir str | pathlib.Path models 保存和加载模型的子目录
wd float | int | None None 默认权重衰减
wd_bn_bias bool False 对归一化和偏置参数应用权重衰减
train_bn bool True 训练冻结的归一化层
moms tuple (0.95, 0.85, 0.95) 调度器的默认动量
default_cbs bool True 包含默认的 Callbacks

源代码

VerboseCallback

 VerboseCallback (after_create=None, before_fit=None, before_epoch=None,
                  before_train=None, before_batch=None, after_pred=None,
                  after_loss=None, before_backward=None,
                  after_cancel_backward=None, after_backward=None,
                  before_step=None, after_cancel_step=None,
                  after_step=None, after_cancel_batch=None,
                  after_batch=None, after_cancel_train=None,
                  after_train=None, before_validate=None,
                  after_cancel_validate=None, after_validate=None,
                  after_cancel_epoch=None, after_epoch=None,
                  after_cancel_fit=None, after_fit=None)

打印每个被调用的事件名称的回调函数

安装工具

源代码

get_env

 get_env (name)

如果定义了环境变量且非空，则返回其值，否则返回 Unknown

源代码

try_import

 try_import (module)

尝试导入 module。成功则返回模块对象，失败则返回 None

源代码

nvidia_smi

 nvidia_smi (cmd='nvidia-smi')
res = nvidia_smi()

源代码

nvidia_mem

 nvidia_mem ()
nvidia_mem()
['48600', '7982']

源代码

show_install

 show_install (show_nvidia_smi:bool=False)

打印用户设置信息

- 导出