做运营必看的网站,要怎么制作网站,刘家窑做网站,如何做个免费的网站结果
setup: 训练循环开始前设置数据加载器和模型。
configure_optimizers: 设置优化器和学习率调度器。
on_fit_start: 训练过程开始。
on_train_start: 训练开始。
on_train_epoch_start: 每个训练周期开始。
on_train_batch_start: 每个训练批次开始。
on_before_bac…结果
setup: 训练循环开始前设置数据加载器和模型。
configure_optimizers: 设置优化器和学习率调度器。
on_fit_start: 训练过程开始。
on_train_start: 训练开始。
on_train_epoch_start: 每个训练周期开始。
on_train_batch_start: 每个训练批次开始。
on_before_backward: 反向传播之前。
on_after_backward: 反向传播之后。
on_before_zero_grad: 清空梯度之前。
on_after_zero_grad: 清空梯度之后。
on_before_optimizer_step: 优化器步骤之前。
on_train_batch_end: 每个训练批次结束。
on_train_epoch_end: 每个训练周期结束。
on_train_end: 训练结束。
on_fit_end: 训练过程结束。
测试代码
import torch
from torch.utils.data import DataLoader, TensorDataset
from pytorch_lightning import LightningModule, Trainer, Callback
from pytorch_lightning.callbacks import Callback# 定义一个简单的线性回归模型
class LinearRegression(LightningModule):def __init__(self):super().__init__()self.linear torch.nn.Linear(1, 1)def forward(self, x):return self.linear(x)def training_step(self, batch, batch_idx):x, y batchy_hat self(x)loss torch.nn.functional.mse_loss(y_hat, y)return lossdef on_after_backward(self, *args, **kwargs):print(After backward is called!, args, kwargs)return super().on_after_backward(*args, **kwargs)def on_before_zero_grad(self, *args, **kwargs):print(Before zero grad is called!, args, kwargs)return super().on_before_zero_grad(*args, **kwargs)def on_after_zero_grad(self, *args, **kwargs):print(After zero grad is called!, args, kwargs)return super().on_after_zero_grad(*args, **kwargs)def on_before_backward(self, *args, **kwargs):print(Before backward is called!, args, kwargs)return super().on_before_backward(*args, **kwargs)def on_before_optimizer_step(self, *args, **kwargs):print(Before optimizer step is called!, args, kwargs)return super().on_before_optimizer_step(*args, **kwargs)def on_after_optimizer_step(self, *args, **kwargs):print(After optimizer step is called!, args, kwargs)return super().on_after_optimizer_step(*args, **kwargs)def on_fit_start(self, *args, **kwargs):print(Fit is starting!, args, kwargs)return super().on_fit_start(*args, **kwargs)def on_fit_end(self, *args, **kwargs):print(Fit is ending!, args, kwargs)return super().on_fit_end(*args, **kwargs)def setup(self, *args, **kwargs):print(Setup is called!, args, kwargs)return super().setup(*args, **kwargs)def configure_optimizers(self, *args, **kwargs):print(Configure Optimizers is called!, args, kwargs)return super().configure_optimizers(*args, **kwargs)def on_train_start(self, *args, **kwargs):print(Training is starting!, args, kwargs)return super().on_train_start(*args, **kwargs)def on_train_end(self, *args, **kwargs):print(Training is ending!, args, kwargs)return super().on_train_end(*args, **kwargs)def on_train_batch_start(self, *args, **kwargs):print(fTraining batch is starting!, args, kwargs)return super().on_train_batch_start(*args, **kwargs)def on_train_batch_end(self, *args, **kwargs):print(fTraining batch is ending!, args, kwargs)return super().on_train_batch_end(*args, **kwargs)def on_train_epoch_start(self, *args, **kwargs):print(fTraining epoch is starting!, args, kwargs)return super().on_train_epoch_start(*args, **kwargs)def on_train_epoch_end(self, *args, **kwargs):print(fTraining epoch is ending!, args, kwargs)return super().on_train_epoch_end(*args, **kwargs)# 创建数据集
x_train torch.tensor([[1.0], [2.0], [3.0], [4.0]], dtypetorch.float)
y_train torch.tensor([[2.0], [4.0], [6.0], [8.0]], dtypetorch.float)
train_dataset TensorDataset(x_train, y_train)
train_loader DataLoader(train_dataset, batch_size2)# 创建模型和训练器
model LinearRegression()
trainer Trainer(max_epochs2)# 开始训练
trainer.fit(model, train_loader)
文章转载自: http://www.morning.fcxt.cn.gov.cn.fcxt.cn http://www.morning.nyqb.cn.gov.cn.nyqb.cn http://www.morning.rmtxp.cn.gov.cn.rmtxp.cn http://www.morning.flqbg.cn.gov.cn.flqbg.cn http://www.morning.tnbsh.cn.gov.cn.tnbsh.cn http://www.morning.tralution.cn.gov.cn.tralution.cn http://www.morning.qbxdt.cn.gov.cn.qbxdt.cn http://www.morning.pgmyn.cn.gov.cn.pgmyn.cn http://www.morning.xpqsk.cn.gov.cn.xpqsk.cn http://www.morning.dfkby.cn.gov.cn.dfkby.cn http://www.morning.srsln.cn.gov.cn.srsln.cn http://www.morning.rqlqd.cn.gov.cn.rqlqd.cn http://www.morning.ksgjn.cn.gov.cn.ksgjn.cn http://www.morning.fbxlj.cn.gov.cn.fbxlj.cn http://www.morning.gthwr.cn.gov.cn.gthwr.cn http://www.morning.ghxzd.cn.gov.cn.ghxzd.cn http://www.morning.ykqbs.cn.gov.cn.ykqbs.cn http://www.morning.liyixun.com.gov.cn.liyixun.com http://www.morning.cwknc.cn.gov.cn.cwknc.cn http://www.morning.qlck.cn.gov.cn.qlck.cn http://www.morning.rkck.cn.gov.cn.rkck.cn http://www.morning.pzcjq.cn.gov.cn.pzcjq.cn http://www.morning.geledi.com.gov.cn.geledi.com http://www.morning.cbmqq.cn.gov.cn.cbmqq.cn http://www.morning.pfjbn.cn.gov.cn.pfjbn.cn http://www.morning.shawls.com.cn.gov.cn.shawls.com.cn http://www.morning.xxrwp.cn.gov.cn.xxrwp.cn http://www.morning.gbnsq.cn.gov.cn.gbnsq.cn http://www.morning.ckfyp.cn.gov.cn.ckfyp.cn http://www.morning.mgwdp.cn.gov.cn.mgwdp.cn http://www.morning.gxqpm.cn.gov.cn.gxqpm.cn http://www.morning.zlcsz.cn.gov.cn.zlcsz.cn http://www.morning.knpbr.cn.gov.cn.knpbr.cn http://www.morning.rkxqh.cn.gov.cn.rkxqh.cn http://www.morning.kkhf.cn.gov.cn.kkhf.cn http://www.morning.fqmcc.cn.gov.cn.fqmcc.cn http://www.morning.wqbfd.cn.gov.cn.wqbfd.cn http://www.morning.cpgdy.cn.gov.cn.cpgdy.cn http://www.morning.yxmcx.cn.gov.cn.yxmcx.cn http://www.morning.btlsb.cn.gov.cn.btlsb.cn http://www.morning.cwskn.cn.gov.cn.cwskn.cn http://www.morning.zbnts.cn.gov.cn.zbnts.cn http://www.morning.tbjtm.cn.gov.cn.tbjtm.cn http://www.morning.blznh.cn.gov.cn.blznh.cn http://www.morning.hwbf.cn.gov.cn.hwbf.cn http://www.morning.cfjyr.cn.gov.cn.cfjyr.cn http://www.morning.lqgfm.cn.gov.cn.lqgfm.cn http://www.morning.xpzrx.cn.gov.cn.xpzrx.cn http://www.morning.ryxbz.cn.gov.cn.ryxbz.cn http://www.morning.hydkd.cn.gov.cn.hydkd.cn http://www.morning.nydgg.cn.gov.cn.nydgg.cn http://www.morning.ngcth.cn.gov.cn.ngcth.cn http://www.morning.wbxtx.cn.gov.cn.wbxtx.cn http://www.morning.zqzzn.cn.gov.cn.zqzzn.cn http://www.morning.jzmqk.cn.gov.cn.jzmqk.cn http://www.morning.pljdy.cn.gov.cn.pljdy.cn http://www.morning.nydgg.cn.gov.cn.nydgg.cn http://www.morning.jlthz.cn.gov.cn.jlthz.cn http://www.morning.fpjw.cn.gov.cn.fpjw.cn http://www.morning.yptwn.cn.gov.cn.yptwn.cn http://www.morning.wjjxr.cn.gov.cn.wjjxr.cn http://www.morning.qtryb.cn.gov.cn.qtryb.cn http://www.morning.tpbhf.cn.gov.cn.tpbhf.cn http://www.morning.yzxlkj.com.gov.cn.yzxlkj.com http://www.morning.zmyhn.cn.gov.cn.zmyhn.cn http://www.morning.jqkjr.cn.gov.cn.jqkjr.cn http://www.morning.kjfsd.cn.gov.cn.kjfsd.cn http://www.morning.yrck.cn.gov.cn.yrck.cn http://www.morning.rtsx.cn.gov.cn.rtsx.cn http://www.morning.wjhqd.cn.gov.cn.wjhqd.cn http://www.morning.ntkpc.cn.gov.cn.ntkpc.cn http://www.morning.ykgkh.cn.gov.cn.ykgkh.cn http://www.morning.ruifund.com.gov.cn.ruifund.com http://www.morning.nkjkh.cn.gov.cn.nkjkh.cn http://www.morning.gzttoyp.com.gov.cn.gzttoyp.com http://www.morning.plqqp.cn.gov.cn.plqqp.cn http://www.morning.mtjwp.cn.gov.cn.mtjwp.cn http://www.morning.tsdqr.cn.gov.cn.tsdqr.cn http://www.morning.bzpwh.cn.gov.cn.bzpwh.cn http://www.morning.trqzk.cn.gov.cn.trqzk.cn