reworked network arch diagrams

This commit is contained in:
Jan Kowalczyk
2025-09-01 18:53:01 +02:00
parent 5ff56994c0
commit 33de01b150
17 changed files with 156 additions and 36 deletions

View File

@@ -38,6 +38,7 @@ arch = [
to_fc(
"fc3",
n_filer="{{4×512×8}}",
zlabeloffset=0.35,
offset="(2,0,0)",
to="(latent-east)",
height=1.3,
@@ -49,6 +50,7 @@ arch = [
# Reshape to 4×8×512
to_UnPool(
"up1",
n_filer=4,
offset="(2,0,0)",
to="(fc3-east)",
height=H16,
@@ -61,6 +63,7 @@ arch = [
to_Conv(
"deconv1",
s_filer="{{1024×16}}",
zlabeloffset=0.2,
n_filer=8,
offset="(0,0,0)",
to="(up1-east)",
@@ -74,6 +77,7 @@ arch = [
to_UnPool(
"up2",
offset="(2,0,0)",
n_filer=8,
to="(deconv1-east)",
height=H32,
depth=D2048,
@@ -85,6 +89,7 @@ arch = [
to_Conv(
"deconv2",
s_filer="{{2048×32}}",
zlabeloffset=0.15,
n_filer=1,
offset="(0,0,0)",
to="(up2-east)",
@@ -98,6 +103,7 @@ arch = [
to_Conv(
"out",
s_filer="{{2048×32}}",
zlabeloffset=0.15,
n_filer=1,
offset="(2,0,0)",
to="(deconv2-east)",