parent
2ae37a4ea2
commit
2e97045c23
@ -0,0 +1,50 @@
|
|||||||
|
from paddle.trainer_config_helpers import *
|
||||||
|
|
||||||
|
settings(batch_size=1000, learning_rate=1e-5)
|
||||||
|
|
||||||
|
num_channels = 3
|
||||||
|
filter_size = 3
|
||||||
|
filter_size_y = 3
|
||||||
|
filter_size_z = 3
|
||||||
|
stride = 2
|
||||||
|
stride_y = 2
|
||||||
|
stride_z = 2
|
||||||
|
padding = 1
|
||||||
|
padding_y = 1
|
||||||
|
padding_z = 1
|
||||||
|
groups = 1
|
||||||
|
|
||||||
|
data = data_layer(
|
||||||
|
name='data', size=12096 * num_channels, height=48, width=42, depth=6)
|
||||||
|
|
||||||
|
# first
|
||||||
|
deconv3d_1 = img_conv3d_layer(
|
||||||
|
input=data,
|
||||||
|
name='deconv3d_1',
|
||||||
|
num_filters=16,
|
||||||
|
num_channels=num_channels,
|
||||||
|
filter_size=filter_size,
|
||||||
|
stride=stride,
|
||||||
|
padding=padding,
|
||||||
|
groups=groups,
|
||||||
|
bias_attr=True,
|
||||||
|
shared_biases=True,
|
||||||
|
trans=True,
|
||||||
|
layer_type="deconv3d",
|
||||||
|
act=LinearActivation())
|
||||||
|
# second
|
||||||
|
deconv3d_2 = img_conv3d_layer(
|
||||||
|
input=data,
|
||||||
|
name='deconv3d_2',
|
||||||
|
num_filters=16,
|
||||||
|
num_channels=num_channels,
|
||||||
|
filter_size=[filter_size, filter_size_y, filter_size_z],
|
||||||
|
stride=[stride, stride_y, stride_z],
|
||||||
|
padding=[padding, padding_y, padding_z],
|
||||||
|
groups=groups,
|
||||||
|
bias_attr=True,
|
||||||
|
shared_biases=True,
|
||||||
|
trans=True,
|
||||||
|
layer_type="deconv3d",
|
||||||
|
act=LinearActivation())
|
||||||
|
outputs(deconv3d_2)
|
Loading…
Reference in new issue