202 lines
11 KiB
Plaintext
202 lines
11 KiB
Plaintext
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
|