Получить входные каналы для conv2d с предыдущего слоя?

Мне было интересно, много ли сверточных слоев (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?

См. также:  Файл дампа ядра не создается после ошибки сегментации в macOS Big Sur
Понравилась статья? Поделиться с друзьями:
IT Шеф
Комментарии: 1
  1. hsuyaa

    Второй параметр конструктора nn.Conv2D — количество выходных каналов:

    self.conv1 = nn.Conv2D(in_channels,conv1_out_channels)
    self.conv2 = nn.Conv2D(conv1_out_channels,out_ch2)
    

    как описано в документации

    Также доступны как недвижимость:

    self.conv1.out_channels
    

    Антон Спасибо. Я получил то, что искал, на странице обсуждений pytorch. См. это </ а> person hsuyaa; 27.04.2020

Добавить комментарий

;-) :| :x :twisted: :smile: :shock: :sad: :roll: :razz: :oops: :o :mrgreen: :lol: :idea: :grin: :evil: :cry: :cool: :arrow: :???: :?: :!: