reworked network arch diagrams

thesis/third_party/PlotNeuralNet/deepsad/arch_ef_decoder.py

@@ -38,6 +38,7 @@ arch = [
     to_fc(
         "fc3",
         n_filer="{{8×128×8}}",
+        zlabeloffset=0.5,
         offset="(2,0,0)",
         to="(latent-east)",
         height=H1,
@@ -47,20 +48,22 @@ arch = [
     ),
     to_Conv(
         "unsqueeze",
-        s_filer="",
+        s_filer="{{128×8}}",
+        zlabeloffset=0.4,
         n_filer=32,
         offset="(2,0,0)",
         to="(fc3-east)",
         height=H8,
         depth=D128,
         width=W32,
-        caption="unsqueeze",
+        caption="Unsqueeze",
     ),
     # to_connection("latent", "fc3"),
     # Reshape to 4×8×512
     to_UnPool(
         "up1",
         offset="(2,0,0)",
+        n_filer=32,
         to="(unsqueeze-east)",
         height=H16,
         depth=D256,
@@ -76,11 +79,12 @@ arch = [
         height=H16,
         depth=D256,
         width=W1,
-        caption="deconv1",
+        caption="Deconv1",
     ),
     to_Conv(
         "dwdeconv2",
         s_filer="{{256×16}}",
+        zlabeloffset=0.4,
         n_filer=32,
         offset="(0,0,0)",
         to="(dwdeconv1-east)",
@@ -93,6 +97,7 @@ arch = [
         "up2",
         offset="(2,0,0)",
         to="(dwdeconv2-east)",
+        n_filer=32,
         height=H16,
         depth=D1024,
         width=W32,
@@ -107,11 +112,12 @@ arch = [
         height=H16,
         depth=D1024,
         width=W1,
-        caption="deconv2",
+        caption="Deconv2",
     ),
     to_Conv(
         "dwdeconv4",
         s_filer="{{1024×16}}",
+        zlabeloffset=0.17,
         n_filer=16,
         offset="(0,0,0)",
         to="(dwdeconv3-east)",
@@ -123,6 +129,7 @@ arch = [
     to_UnPool(
         "up3",
         offset="(2,0,0)",
+        n_filer=16,
         to="(dwdeconv4-east)",
         height=H32,
         depth=D2048,
@@ -138,11 +145,12 @@ arch = [
         height=H32,
         depth=D2048,
         width=W1,
-        caption="deconv3",
+        caption="Deconv3",
     ),
     to_Conv(
         "dwdeconv6",
         s_filer="{{2048×32}}",
+        zlabeloffset=0.15,
         n_filer=8,
         offset="(0,0,0)",
         to="(dwdeconv5-east)",
@@ -154,26 +162,28 @@ arch = [
     to_Conv(
         "outconv",
         s_filer="{{2048×32}}",
+        zlabeloffset=0.15,
         n_filer=1,
         offset="(2,0,0)",
         to="(dwdeconv6-east)",
         height=H32,
         depth=D2048,
         width=W1,
-        caption="deconv4",
+        caption="Deconv4",
     ),
     # to_connection("up2", "deconv2"),
     # Output
     to_Conv(
         "out",
         s_filer="{{2048×32}}",
+        zlabeloffset=0.15,
         n_filer=1,
         offset="(2,0,0)",
         to="(outconv-east)",
         height=H32,
         depth=D2048,
         width=W1,
-        caption="output",
+        caption="Output",
     ),
     # to_connection("deconv2", "out"),
     to_end(),