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

@@ -27,8 +27,9 @@
\pic[shift={(0,0,0)}] at (0,0,0)
{Box={
name=input,
caption=input,
caption=Input,
xlabel={{1, }},
zlabeloffset=0.2,
zlabel={{2048×32}},
fill=\ConvColor,
height=26,
@@ -43,6 +44,7 @@
name=dwconv1,
caption=,
xlabel={{1, }},
zlabeloffset=0.3,
zlabel=,
fill=\ConvColor,
height=26,
@@ -55,9 +57,10 @@
\pic[shift={(0,0,0)}] at (dwconv1-east)
{Box={
name=dwconv2,
caption=conv1,
caption=Conv1,
xlabel={{16, }},
zlabel=,
zlabeloffset=0.15,
zlabel={{2048×32}},
fill=\ConvColor,
height=26,
width=4,
@@ -69,6 +72,9 @@
\pic[shift={ (0,0,0) }] at (dwconv2-east)
{Box={
name=pool1,
xlabel={{, }},
zlabeloffset=0.3,
zlabel={{512×32}},
caption=,
fill=\PoolColor,
opacity=0.5,
@@ -84,6 +90,7 @@
name=dwconv3,
caption=,
xlabel={{1, }},
zlabeloffset=0.3,
zlabel=,
fill=\ConvColor,
height=26,
@@ -96,9 +103,10 @@
\pic[shift={(0,0,0)}] at (dwconv3-east)
{Box={
name=dwconv4,
caption=conv2,
caption=Conv2,
xlabel={{32, }},
zlabel=,
zlabeloffset=0.3,
zlabel={{512×32}},
fill=\ConvColor,
height=26,
width=8,
@@ -110,6 +118,9 @@
\pic[shift={ (0,0,0) }] at (dwconv4-east)
{Box={
name=pool2,
xlabel={{, }},
zlabeloffset=0.45,
zlabel={{256×16}},
caption=,
fill=\PoolColor,
opacity=0.5,
@@ -123,6 +134,9 @@
\pic[shift={ (0,0,0) }] at (pool2-east)
{Box={
name=pool3,
xlabel={{, }},
zlabeloffset=0.45,
zlabel={{128×8}},
caption=,
fill=\PoolColor,
opacity=0.5,
@@ -136,9 +150,10 @@
\pic[shift={(2,0,0)}] at (pool3-east)
{Box={
name=squeeze,
caption=squeeze,
caption=Squeeze,
xlabel={{8, }},
zlabel=,
zlabeloffset=0.45,
zlabel={{128×8}},
fill=\ConvColor,
height=12,
width=2,
@@ -152,6 +167,7 @@
name=fc1,
caption=FC,
xlabel={{" ","dummy"}},
zlabeloffset=0.5,
zlabel={{8×128×8}},
fill=\FcColor,
opacity=0.8,
@@ -167,6 +183,7 @@
name=latent,
caption=Latent Space,
xlabel={{, }},
zlabeloffset=0.3,
zlabel=latent dim,
fill=\ConvColor,
height=19.200000000000003,