Мне было интересно, много ли сверточных слоев (conv1 -> conv2). Как мы можем получить параметр входных каналов для conv2 из выходного канала conv1?
class MyModel(nn.Module):
def __init__(self, in_ch, num_features, out_ch2):
super(MyModel, self).__init__()
self.conv1 = nn.Conv2D(in_channels,num_features)
self.conv2 = nn.Conv2D(in_channnels_from_out_channels_of_conv1,out_ch2)
Могу ли я получить out_channels из слоя conv1 и использовать его как in_ch для conv2?
Второй параметр конструктора nn.Conv2D — количество выходных каналов:
как описано в документации
Также доступны как недвижимость:
Антон Спасибо. Я получил то, что искал, на странице обсуждений pytorch. См. это </ а> — person hsuyaa; 27.04.2020