Как я могу написать модель Pytorch с использованием алмазного наследования? ⇐ Python

[Anonymous]

Мне нужно реализовать сложную модель, и мне лучше использовать наследование Diamond для удовлетворения различных требований. Вот игрушечный код, показывающий, что я пытаюсь сделать:

Код: Выделить всё

import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.utils.model_zoo as modelzoo
import torch.distributed as dist

class A(nn.Module):

def __init__(self, ratio=4, *args, **kwargs):
super().__init__()

self.conv_base = nn.Conv2d(3, 3 * ratio, 3, 1, 1)

class B(A):

def __init__(self, b_args, **kwargs):
super().__init__(ratio=4)

self.conv1 = nn.Conv2d(4, 3, 1, 1, 0)

class C(A):

def __init__(self, c_args, **kwargs):
super().__init__(ratio=4)

self.conv2 = nn.Conv2d(4, 3, 1, 1, 0)

class D(B, C):

def __init__(self, b_args, c_args):
super().__init__(b_args=b_args, c_args=c_args)

self.conv3 = nn.Conv2d(4, 3, 1, 1, 0)

b_args = dict(a=1)
c_args = dict(b=2)
model = D(b_args, c_args)

Я получил ошибку:

Код: Выделить всё

Traceback (most recent call last):
File "tmp.py", line 95, in 
model = D(b_args, c_args)
File "tmp.py", line 88, in __init__
super().__init__(b_args=b_args, c_args=c_args)
File "tmp.py", line 72, in __init__
super().__init__(ratio=4)
TypeError: C.__init__() missing 1 required positional argument: 'c_args'

Скажите, пожалуйста, как мне это сделать?
Кстати, мне нужны B и C быть работоспособными, а это означает, что они являются не только родителями D, но и классом, экземпляр которого можно создать.

Подробнее здесь: https://stackoverflow.com/questions/792 ... itance-way