torchvision 0.15.1 旧版中ConvBNActivation类可以使用什么替换
时间: 2023-06-12 19:07:03 浏览: 110
jetson上torchvision-0.15.1源码
在torchvision 0.15.1中,ConvBNActivation类可以被替换为torch.nn.Sequential()。
可以使用以下代码进行替换:
```python
import torch.nn as nn
# 使用ConvBNActivation类的代码
conv_bn_act = nn.Sequential(
nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, bias=False),
nn.BatchNorm2d(out_channels),
nn.ReLU(inplace=True)
)
# 替换为torch.nn.Sequential()的代码
conv_bn_act = nn.Sequential(
nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, bias=False),
nn.BatchNorm2d(out_channels),
nn.ReLU(inplace=True)
)
```
其中,in_channels、out_channels和kernel_size需要根据具体的情况进行替换。
阅读全文