BNReLU2d
-
class torch.ao.nn.intrinsic.quantized.BNReLU2d(num_features, eps=1e-05, momentum=0.1, device=None, dtype=None)[source][source]
BNReLU2d 模块是 BatchNorm2d 和 ReLU 的融合模块
我们采用与 torch.ao.nn.quantized.BatchNorm2d
相同的接口。
- 变量:
torch.ao.nn.quantized.BatchNorm2d(相同于)–