Importing module 'MDX_Net' from 'modules'...
Successfully imported class 'MDX_Net'.
--- PyTorch Model Structure ---
model.MDX_Net: MDX_Net
  model.initial_conv: Conv2d
  model.initial_relu: ReLU
  model.initial_transpose: Transpose
  model.enc_stages.ModuleList: ModuleList
    model.enc_stages.0.TDF_Block: TDF_Block
      model.enc_stages.0.time_branch.TimeBranch: TimeBranch
          model.enc_stages.0.time_branch.sequence.0: Conv2d
          model.enc_stages.0.time_branch.sequence.1: ReLU
          model.enc_stages.0.time_branch.sequence.2: Conv2d
          model.enc_stages.0.time_branch.sequence.3: ReLU
          model.enc_stages.0.time_branch.sequence.4: Conv2d
          model.enc_stages.0.time_branch.sequence.5: ReLU
      model.enc_stages.0.freq_branch.FrequencyBranch: FrequencyBranch
          model.enc_stages.0.freq_branch.sequence.0: Linear
          model.enc_stages.0.freq_branch.sequence.1: BatchNorm2d
          model.enc_stages.0.freq_branch.sequence.2: ReLU
          model.enc_stages.0.freq_branch.sequence.3: Linear
          model.enc_stages.0.freq_branch.sequence.4: BatchNorm2d
          model.enc_stages.0.freq_branch.sequence.5: ReLU
      model.enc_stages.1.0: Conv2d
      model.enc_stages.1.1: ReLU
    model.enc_stages.2.TDF_Block: TDF_Block
      model.enc_stages.2.time_branch.TimeBranch: TimeBranch
          model.enc_stages.2.time_branch.sequence.0: Conv2d
          model.enc_stages.2.time_branch.sequence.1: ReLU
          model.enc_stages.2.time_branch.sequence.2: Conv2d
          model.enc_stages.2.time_branch.sequence.3: ReLU
          model.enc_stages.2.time_branch.sequence.4: Conv2d
          model.enc_stages.2.time_branch.sequence.5: ReLU
      model.enc_stages.2.freq_branch.FrequencyBranch: FrequencyBranch
          model.enc_stages.2.freq_branch.sequence.0: Linear
          model.enc_stages.2.freq_branch.sequence.1: BatchNorm2d
          model.enc_stages.2.freq_branch.sequence.2: ReLU
          model.enc_stages.2.freq_branch.sequence.3: Linear
          model.enc_stages.2.freq_branch.sequence.4: BatchNorm2d
          model.enc_stages.2.freq_branch.sequence.5: ReLU
      model.enc_stages.3.0: Conv2d
      model.enc_stages.3.1: ReLU
    model.enc_stages.4.TDF_Block: TDF_Block
      model.enc_stages.4.time_branch.TimeBranch: TimeBranch
          model.enc_stages.4.time_branch.sequence.0: Conv2d
          model.enc_stages.4.time_branch.sequence.1: ReLU
          model.enc_stages.4.time_branch.sequence.2: Conv2d
          model.enc_stages.4.time_branch.sequence.3: ReLU
          model.enc_stages.4.time_branch.sequence.4: Conv2d
          model.enc_stages.4.time_branch.sequence.5: ReLU
      model.enc_stages.4.freq_branch.FrequencyBranch: FrequencyBranch
          model.enc_stages.4.freq_branch.sequence.0: Linear
          model.enc_stages.4.freq_branch.sequence.1: BatchNorm2d
          model.enc_stages.4.freq_branch.sequence.2: ReLU
          model.enc_stages.4.freq_branch.sequence.3: Linear
          model.enc_stages.4.freq_branch.sequence.4: BatchNorm2d
          model.enc_stages.4.freq_branch.sequence.5: ReLU
      model.enc_stages.5.0: Conv2d
      model.enc_stages.5.1: ReLU
    model.enc_stages.6.TDF_Block: TDF_Block
      model.enc_stages.6.time_branch.TimeBranch: TimeBranch
          model.enc_stages.6.time_branch.sequence.0: Conv2d
          model.enc_stages.6.time_branch.sequence.1: ReLU
          model.enc_stages.6.time_branch.sequence.2: Conv2d
          model.enc_stages.6.time_branch.sequence.3: ReLU
          model.enc_stages.6.time_branch.sequence.4: Conv2d
          model.enc_stages.6.time_branch.sequence.5: ReLU
      model.enc_stages.6.freq_branch.FrequencyBranch: FrequencyBranch
          model.enc_stages.6.freq_branch.sequence.0: Linear
          model.enc_stages.6.freq_branch.sequence.1: BatchNorm2d
          model.enc_stages.6.freq_branch.sequence.2: ReLU
          model.enc_stages.6.freq_branch.sequence.3: Linear
          model.enc_stages.6.freq_branch.sequence.4: BatchNorm2d
          model.enc_stages.6.freq_branch.sequence.5: ReLU
      model.enc_stages.7.0: Conv2d
      model.enc_stages.7.1: ReLU
    model.enc_stages.8.TDF_Block: TDF_Block
      model.enc_stages.8.time_branch.TimeBranch: TimeBranch
          model.enc_stages.8.time_branch.sequence.0: Conv2d
          model.enc_stages.8.time_branch.sequence.1: ReLU
          model.enc_stages.8.time_branch.sequence.2: Conv2d
          model.enc_stages.8.time_branch.sequence.3: ReLU
          model.enc_stages.8.time_branch.sequence.4: Conv2d
          model.enc_stages.8.time_branch.sequence.5: ReLU
      model.enc_stages.8.freq_branch.FrequencyBranch: FrequencyBranch
          model.enc_stages.8.freq_branch.sequence.0: Linear
          model.enc_stages.8.freq_branch.sequence.1: BatchNorm2d
          model.enc_stages.8.freq_branch.sequence.2: ReLU
          model.enc_stages.8.freq_branch.sequence.3: Linear
          model.enc_stages.8.freq_branch.sequence.4: BatchNorm2d
          model.enc_stages.8.freq_branch.sequence.5: ReLU
      model.enc_stages.9.0: Conv2d
      model.enc_stages.9.1: ReLU
  model.bottleneck.TDF_Block: TDF_Block
    model.bottleneck.time_branch.TimeBranch: TimeBranch
        model.bottleneck.time_branch.sequence.0: Conv2d
        model.bottleneck.time_branch.sequence.1: ReLU
        model.bottleneck.time_branch.sequence.2: Conv2d
        model.bottleneck.time_branch.sequence.3: ReLU
        model.bottleneck.time_branch.sequence.4: Conv2d
        model.bottleneck.time_branch.sequence.5: ReLU
    model.bottleneck.freq_branch.FrequencyBranch: FrequencyBranch
        model.bottleneck.freq_branch.sequence.0: Linear
        model.bottleneck.freq_branch.sequence.1: BatchNorm2d
        model.bottleneck.freq_branch.sequence.2: ReLU
        model.bottleneck.freq_branch.sequence.3: Linear
        model.bottleneck.freq_branch.sequence.4: BatchNorm2d
        model.bottleneck.freq_branch.sequence.5: ReLU
  model.dec_stages.ModuleList: ModuleList
      model.dec_stages.0.0: ConvTranspose2d
      model.dec_stages.0.1: BatchNorm2d
      model.dec_stages.0.2: ReLU
    model.dec_stages.1.TDF_Block: TDF_Block
      model.dec_stages.1.time_branch.TimeBranch: TimeBranch
          model.dec_stages.1.time_branch.sequence.0: Conv2d
          model.dec_stages.1.time_branch.sequence.1: ReLU
          model.dec_stages.1.time_branch.sequence.2: Conv2d
          model.dec_stages.1.time_branch.sequence.3: ReLU
          model.dec_stages.1.time_branch.sequence.4: Conv2d
          model.dec_stages.1.time_branch.sequence.5: ReLU
      model.dec_stages.1.freq_branch.FrequencyBranch: FrequencyBranch
          model.dec_stages.1.freq_branch.sequence.0: Linear
          model.dec_stages.1.freq_branch.sequence.1: BatchNorm2d
          model.dec_stages.1.freq_branch.sequence.2: ReLU
          model.dec_stages.1.freq_branch.sequence.3: Linear
          model.dec_stages.1.freq_branch.sequence.4: BatchNorm2d
          model.dec_stages.1.freq_branch.sequence.5: ReLU
      model.dec_stages.2.0: ConvTranspose2d
      model.dec_stages.2.1: BatchNorm2d
      model.dec_stages.2.2: ReLU
    model.dec_stages.3.TDF_Block: TDF_Block
      model.dec_stages.3.time_branch.TimeBranch: TimeBranch
          model.dec_stages.3.time_branch.sequence.0: Conv2d
          model.dec_stages.3.time_branch.sequence.1: ReLU
          model.dec_stages.3.time_branch.sequence.2: Conv2d
          model.dec_stages.3.time_branch.sequence.3: ReLU
          model.dec_stages.3.time_branch.sequence.4: Conv2d
          model.dec_stages.3.time_branch.sequence.5: ReLU
      model.dec_stages.3.freq_branch.FrequencyBranch: FrequencyBranch
          model.dec_stages.3.freq_branch.sequence.0: Linear
          model.dec_stages.3.freq_branch.sequence.1: BatchNorm2d
          model.dec_stages.3.freq_branch.sequence.2: ReLU
          model.dec_stages.3.freq_branch.sequence.3: Linear
          model.dec_stages.3.freq_branch.sequence.4: BatchNorm2d
          model.dec_stages.3.freq_branch.sequence.5: ReLU
      model.dec_stages.4.0: ConvTranspose2d
      model.dec_stages.4.1: BatchNorm2d
      model.dec_stages.4.2: ReLU
    model.dec_stages.5.TDF_Block: TDF_Block
      model.dec_stages.5.time_branch.TimeBranch: TimeBranch
          model.dec_stages.5.time_branch.sequence.0: Conv2d
          model.dec_stages.5.time_branch.sequence.1: ReLU
          model.dec_stages.5.time_branch.sequence.2: Conv2d
          model.dec_stages.5.time_branch.sequence.3: ReLU
          model.dec_stages.5.time_branch.sequence.4: Conv2d
          model.dec_stages.5.time_branch.sequence.5: ReLU
      model.dec_stages.5.freq_branch.FrequencyBranch: FrequencyBranch
          model.dec_stages.5.freq_branch.sequence.0: Linear
          model.dec_stages.5.freq_branch.sequence.1: BatchNorm2d
          model.dec_stages.5.freq_branch.sequence.2: ReLU
          model.dec_stages.5.freq_branch.sequence.3: Linear
          model.dec_stages.5.freq_branch.sequence.4: BatchNorm2d
          model.dec_stages.5.freq_branch.sequence.5: ReLU
      model.dec_stages.6.0: ConvTranspose2d
      model.dec_stages.6.1: BatchNorm2d
      model.dec_stages.6.2: ReLU
    model.dec_stages.7.TDF_Block: TDF_Block
      model.dec_stages.7.time_branch.TimeBranch: TimeBranch
          model.dec_stages.7.time_branch.sequence.0: Conv2d
          model.dec_stages.7.time_branch.sequence.1: ReLU
          model.dec_stages.7.time_branch.sequence.2: Conv2d
          model.dec_stages.7.time_branch.sequence.3: ReLU
          model.dec_stages.7.time_branch.sequence.4: Conv2d
          model.dec_stages.7.time_branch.sequence.5: ReLU
      model.dec_stages.7.freq_branch.FrequencyBranch: FrequencyBranch
          model.dec_stages.7.freq_branch.sequence.0: Linear
          model.dec_stages.7.freq_branch.sequence.1: BatchNorm2d
          model.dec_stages.7.freq_branch.sequence.2: ReLU
          model.dec_stages.7.freq_branch.sequence.3: Linear
          model.dec_stages.7.freq_branch.sequence.4: BatchNorm2d
          model.dec_stages.7.freq_branch.sequence.5: ReLU
      model.dec_stages.8.0: ConvTranspose2d
      model.dec_stages.8.1: BatchNorm2d
      model.dec_stages.8.2: ReLU
    model.dec_stages.9.TDF_Block: TDF_Block
      model.dec_stages.9.time_branch.TimeBranch: TimeBranch
          model.dec_stages.9.time_branch.sequence.0: Conv2d
          model.dec_stages.9.time_branch.sequence.1: ReLU
          model.dec_stages.9.time_branch.sequence.2: Conv2d
          model.dec_stages.9.time_branch.sequence.3: ReLU
          model.dec_stages.9.time_branch.sequence.4: Conv2d
          model.dec_stages.9.time_branch.sequence.5: ReLU
      model.dec_stages.9.freq_branch.FrequencyBranch: FrequencyBranch
          model.dec_stages.9.freq_branch.sequence.0: Linear
          model.dec_stages.9.freq_branch.sequence.1: BatchNorm2d
          model.dec_stages.9.freq_branch.sequence.2: ReLU
          model.dec_stages.9.freq_branch.sequence.3: Linear
          model.dec_stages.9.freq_branch.sequence.4: BatchNorm2d
          model.dec_stages.9.freq_branch.sequence.5: ReLU
  model.final_transpose: Transpose
  model.final_conv: Conv2d