快捷方式

ConvBnReLU1d

class torch.ao.nn.intrinsic.qat.ConvBnReLU1d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=None, padding_mode='zeros', eps=1e-05, momentum=0.1, freeze_bn=False, qconfig=None)[source][source]

ConvBnReLU1d 模組是一個由 Conv1d、BatchNorm1d 和 ReLU 融合而成的模組,附帶用於權重的 FakeQuantize 模組,用於量化感知訓練。

我們結合了 torch.nn.Conv1dtorch.nn.BatchNorm1dtorch.nn.ReLU 的介面。

torch.nn.Conv1d 類似,FakeQuantize 模組初始化為預設值。

變數

weight_fake_quant – 權重的 fake quant 模組

文件

查閱 PyTorch 的全面開發者文件

查閱文件

教程

獲取針對初學者和高階開發者的深度教程

查閱教程

資源

查詢開發資源並解答您的疑問

查閱資源