Closed blythed closed 2 years ago
PR https://github.com/lf1-io/padl/pull/376 fixes this
@transform
class Generator:
def __init__(self, channel_number=96,
network_depth=5,
in_channels=50,
convolution_type='regular',
image_channels=3):
self.channel_number = channel_number
self.network_depth = network_depth
self.convolution_type = convolution_type
self.image_channels = image_channels
def __call__(self, input):
return input
f = identity - 'f'
gen = Generator(channel_number=96,
network_depth=5,
in_channels=50,
convolution_type='regular',
image_channels=3)
comp = (
f
>> batch
>> f
>> gen
)
Compose - "comp":
β
βΌ args
0: Identity(-?-)
β
βΌ args
1: Batchify(dim=0)
β
βΌ args
2: Identity(-?-)
β
βΌ input
3: Generator(channel_number=96, network_depth=5, in_channels=50, convolution_type='regular', image_channels=3)
comp = (
f
>> batch
>> f + f
>> gen
)
Compose - "comp":
β
βΌ args
0: Identity(-?-)
β
βΌ args
1: Batchify(dim=0)
ββββββββββββββββββββββββββββββββββββββββββββββββββββββ
β β
βΌ args βΌ args
2: Identity(-?-) + Identity(-?-)
β
βΌ input
3: Generator(channel_number=96, network_depth=5, ...)
comp = (
f
>> batch
>> f + f + f
>> gen
)
Compose - "comp":
β
βΌ args
0: Identity(-?-)
β
βΌ args
1: Batchify(dim=0)
βββββββββββββββββββββββββββββββββββββββββββββββββββββ
βββββββββββββββββββββββββββββββββββββ β
β β β
βΌ args βΌ args βΌ args
2: Identity(-?-) + Identity(-?-) + Identity(-?-)
β
βΌ input
3: Generator(channel_number=96, ...)
π°οΈ Feature
Don't want this type of thing.
Signatures with many parameters should be wrapped over several lines.